Machine Learning for Water Quality Index Forecasting

Abstract

This study aims to forecast water quality in the Tumkur district, Karnataka state, India, to increase pollution levels. Various machine learning techniques, including support vector machines, regression trees, linear regression, and neural networks, are employed. The Water Quality Index (WQI) is determined using parameters such as total hardness, pH, alkalinity, turbidity, chloride, dissolved solids, and conductivity. The dataset is split into training and testing sets (80:20) to assess model performance. Support Vector Machines and Linear Regression outperform other models, achieving R2 values of 0.96 and 0.99 for training and testing, respectively. This research underscores the importance of advanced machine learning techniques for accurate water quality prediction, crucial for effective pollution reduction strategies in the region.