Abstract:Water quality is very dominant for humans, animals, plants, industries, and the environment. In the last decades, the quality of water has been impacted by contamination and pollution. In this paper, the challenge is to anticipate Water Quality Index (WQI) and Water Quality Classification (WQC), such that WQI is a vital indicator for water validity. In this study, parameters optimization and tuning are utilized to improve the accuracy of several machine learning models, where the machine learning techniques are utilized for the process of predicting WQI and WQC. Grid search is a vital method used for optimizing and tuning the parameters for four classification models and also, for optimizing and tuning the parameters for four regression models. Random forest (RF) model, Extreme Gradient Boosting (Xgboost) model, Gradient Boosting (GB) model, and Adaptive Boosting (AdaBoost) model are used as classification models for predicting WQC. K-nearest neighbor (KNN) regressor model, decision tree (DT) regressor model, support vector regressor (SVR) model, and multi-layer perceptron (MLP) regressor model are used as regression models for predicting WQI. In addition, preprocessing step including, data imputation (mean imputation) and data normalization were performed to fit the data and make it convenient for any further processing. The dataset used in this study includes 7 features and 1991 instances. To examine the efficacy of the classification approaches, five assessment metrics were computed: accuracy, recall, precision, Matthews's Correlation Coefficient (MCC), and F1 score. To assess the effectiveness of the regression models, four assessment metrics were computed: Mean Absolute Error (MAE), Median Absolute Error (MedAE), Mean Square Error (MSE), and coefficient of determination (R 2 ). In terms of classification, the testing findings showed that the GB model produced the best results, with an accuracy of 99.50% when predicting WQC values. According to the experimental results, the MLP regressor model outperformed other models in regression and achieved an R 2 value of 99.8% while predicting WQI values.

Classification of Salt Quality based on Salt-Forming Composition using Random Forest

Seawater salinity modeling using bivariate probit regression

Quantification of salt stress in wheat leaves by Raman spectroscopy and machine learning

Developing and applying novel spectral feature parameters for classifying soil salt types in arid land

Soil Salinity Mapping with Landsat 8 Oli Imagery and Random Forest Algorithm

The Klasifikasi Kualitas Air Menggunakan Metode KNN, Naïve Bayes, dan Decision Tree

Water quality prediction using machine learning models based on grid search method

Using machine learning architecture to optimize and model the treatment process for saline water level analysis

Spatial Mapping of Soil Salinity Using Machine Learning and Remote Sensing in Kot Addu, Pakistan

Application of fuzzy inference system by Sugeno method on estimating of salt production

Application of Remote-sensing Data and Decision-Tree Analysis to Mapping Salt-Affected Soils over Large Areas

Modeling method of prediction model for salt field ion concentration under solar thermal system using random forest

A Method for Classification of Heavy Mineral Based on Machine Learning.

A Novel Random Forest and its Application on Classification of Air Quality

Qualitative discrimination and quantitative prediction of salt in aqueous solution based on near-infrared spectroscopy

Perbandingan Akurasi, Recall, dan Presisi Klasifikasi pada Algoritma C4.5, Random Forest, SVM dan Naive Bayes

Evaluating machine learning performance in predicting sodium adsorption ratio for sustainable soil-water management in the eastern Mediterranean

Scrutinizing different predictive modeling validation methodologies and data-partitioning strategies: new insights using groundwater modeling case study

Mapping the Levels of Soil Salination and Alkalization by Integrating Machining Learning Methods and Soil-Forming Factors

Estimation of Soil Salt Content at Different Depths Using UAV Multi-Spectral Remote Sensing Combined with Machine Learning Algorithms

Soil salinity prediction and mapping by machine learning regression in Central Mesopotamia, Iraq