Machine-learning tool for researchers and engineers — predict TOC, COD, and UV Absorbance removal efficiency for electrocoagulation (ECP) and polarity-reversal-assisted electrocoagulation (ECP+PR) systems.
Enter your experimental conditions below. The tool uses an Extra Trees ensemble trained on peer-reviewed literature data.
Enter parameters and click Predict to see results
Validated using Group K-Fold cross-validation (grouped by Study_ID to prevent data leakage between studies).
87 observations curated from 9 peer-reviewed studies + in-house experimental data. Covers ECP and ECP+PR systems across municipal wastewater, greywater, textile effluent, landfill leachate, and industrial wastewater.
Extra Trees Regressor ensemble (300 estimators). Five models were trained and compared: Random Forest, Gradient Boosting, Extra Trees, XGBoost, and Voting Ensemble. Extra Trees showed best OOF performance for UV removal.
Group K-Fold cross-validation (5-fold, grouped by Study_ID) prevents data leakage between publications. Both OOF R² (honest generalization) and full-data R² (training fit) are reported.
Feature importance and SHAP analyses identified Treatment Time and Current Density as the dominant predictors. Process type (ECP vs ECP+PR) was also significant — ECP+PR applies periodic polarity reversal to prevent electrode passivation, sustaining higher coagulant generation efficiency throughout the treatment run.