返回
Water quality classification using machine learning algorithms
DOI:10.1016/j.jwpe.2022.102920.png)
摘要
En 中文
Monitoring water quality is essential for protecting human health and the environment and controlling water quality. Artificial Intelligence (AI) offers significant opportunities to help improve the classification and prediction of water quality (WQ). In this study, various AI algorithms are assessed to handle WQ data collected over an extended period and develop a dependable approach for forecasting water quality as accurately as possible. Specifically, various machine learning classifiers and their stacking ensemble models were used to classify the WQ data via the Water Quality Index (WQI). The studied classifiers included Support Vector Machine (SVM), Random Forest (RF), Logistic Regression (LR), Decision Tree (DT), CATBoost, XGBoost, and Multilayer Perceptron (MLP). The dataset used in the study included 1679 samples and their meta-data collected over nine years. In addition, precision-recall curves and Receiver Operating Characteristic curves (ROC) were used to assess the performance of the various classifiers. The findings revealed that the CATBoost model offered the most accurate classifier with a percentage of 94.51. Moreover, after applying stacking ensemble models with all classifiers, accuracy reached 100% in various Meta-classifiers. Furthermore, the CATBoost achieved the highest accuracy as a primary gradient boosting algorithm and a meta classifier. Therefore, the boosting algorithm is proposed as a reliable approach for the WQ classification. The analysis presented in this article presents a framework that can support the efforts of researchers working toward water quality improvement using artificial intelligence.
Keyword:
Water quality
Water quality index
Machine learning
Stack modelling
Meta classifier
Ensemble models
期刊
IF:
6.7
论文数:
1.0W
被引数:
3.3W
机构
引用论文
Evaluation of non-point source pollution and river water quality using a multimedia two-model system
JOURNAL OF HYDROLOGY
IF6.3
A systematic review shows no performance benefit of machine learning over logistic regression for clinical prediction models一项系统评价显示,机器学习在临床预测模型中没有比逻辑回归更多的性能优势

