Water Quality Classification Using Machine Learning Algorithms

Abstract

Protecting and caring for water is one of the most critical environmental problems today. This research aims to design an intelligent system using machine learning models to improve water quality and predict whether it is safe to be used as drinking water. Several models of machine learning algorithms are compared to find the best model to be used for the accuracy of prediction of water quality. In this research, we compare Decision Tree, K-Nearest Neighbor, Support Vector Machine, Ransom Forest, and LightGBM models to get the best model for water potability prediction. Experimental results show that LightGBM model produced the best prediction accuracy of 99.74% on the experimental data.