Over the last few years, different contaminants have posed a danger to the quality of the water. Hence modelling and forecasting water quality are very important in the management of water contamination. The paper proposes an ensemble machine learning-based model for assessing water quality. The results of the proposed model are compared with several machine learning models, including k-nearest neighbour, Naïve Bayes, support vector machine, and decision tree. The considered dataset contains seven statistically important parameters: pH, conductivity, dissolved oxygen, Biochemical Oxygen Demand, nitrate, total coliform, and fecal coliform. The water quality index is calculated for assessing water quality. To utilize an ensemble approach, a voting classifier has been designed with hard voting. The highest prediction accuracy of 99.5% of the water quality index is presented by the voting classifier as compared to the prediction accuracy of 99.2%, 90%, 79%, and 99% presented through k-nearest neighbour, Naïve Bayes, support vector machine, and decision tree, respectively. This was further enhanced to 99.74% using particle swarm based optimization.
scite is a Brooklyn-based organization that helps researchers better discover and understand research articles through Smart Citations–citations that display the context of the citation and describe whether the article provides supporting or contrasting evidence. scite is used by students and researchers from around the world and is funded in part by the National Science Foundation and the National Institute on Drug Abuse of the National Institutes of Health.