Water Quality Index Estimations Using Machine Learning Algorithms: A Case Study of Yazd-Ardakan Plain, Iran

Excessive population growth and high water demands have significantly increased water extractions from deep and semi-deep wells in the arid regions of Iran. This has negatively affected water quality in different areas. The Water Quality Index (WQI) is a suitable tool to assess such impacts. This study used WQI and the fuzzy hierarchical analysis process of the water quality index (FAHP-WQI) to investigate the water quality status of 96 deep agricultural wells in the Yazd-Ardakan Plain, Iran. Calculating the WQI is time-consuming, but estimating WQI is inevitable for water resources management. For this purpose, three Machine Learning (ML) algorithms, namely, Gene Expression Programming (GEP), M5P Model tree, and Multivariate Adaptive Regression Splines (MARS), were employed to predict WQI. Using Wilcox and Schoeller charts, water quality was also investigated for agricultural and drinking purposes. The results demonstrated that 75% and 33% of the study area have good quality, based on the WQI and FAHP-WQI methods, respectively. According to the results of the Wilcox chart, around 37.25% of the wells are in the C3S2 and C3S1 classes, which indicate poor water quality. Schoeller’s diagram placed the drinking water quality of the Yazd-Ardakan plain in acceptable, inadequate, and inappropriate categories. Afterwards, WQI, predicted by means of ML models, were compared on several statistical criteria. Finally, the comparative analysis revealed that MARS is slightly more accurate than the M5P model for estimating WQI.