Water Quality Classification Using SVM And XGBoost Method

• Published in: 2022 IEEE 13th Control and System Graduate Research Colloquium, ICSGRC 2022 - Conference Proceedings
• Main Author: Yusri H.I.H.; Ab Rahim A.A.; Hassan S.L.M.; Halim I.S.A.; Abdullah N.E.
• Format: Conference paper
• Language: English
• Published: Institute of Electrical and Electronics Engineers Inc. 2022
• Online Access: https://www.scopus.com/inward/record.uri?eid=2-s2.0-85137143735&doi=10.1109%2fICSGRC55096.2022.9845143&partnerID=40&md5=c58e052e39c2c218606930b671617786

Various pollutants have been endangering water quality over the past decades. As a result, predicting and modeling water quality have become essential to minimizing water pollution. This research has developed a classification algorithm to predict the water quality classification (WQC). The WQC is classified based on the water quality index (WQI) from 7 parameters in a dataset using Support Vector Machine (SVM) and Extreme Gradient Boosting (XGBoost). The results from the proposed model can accurately classify the water quality based on their features. The research outcome demonstrated that the XGBoost model performed better, with an accuracy of 94%, compared to the SVM model, with only a 67% accuracy. Even better, the XGBoost resulted in only 6% misclassification error compared to SVM, which had 33%. On top of that, XGBoost also obtained consistent superior results from 5-fold validation with an average accuracy of 90%, while SVM with an average accuracy of 64%. Considering the enhanced performance, XGBoost is concluded to be better at water quality classification. © 2022 IEEE.