Comparison of Logistic Regression, Random Forest, SVM, KNN Algorithm for Water Quality Classification Based on Contaminant Parameters

Teguh, Sutanto and Muhammad Rafli, Aditya and Haldi, Budiman and M.Rezqy, Noor Ridha and Usman, Syapotro and Noor, Azijah (2024) Comparison of Logistic Regression, Random Forest, SVM, KNN Algorithm for Water Quality Classification Based on Contaminant Parameters. Journal of Data Science, 2024 (48). pp. 1-7. ISSN 2805-5160

	Text jods2024_48.pdf - Published Version Available under License Creative Commons Attribution. Download (108kB)
	Text 588 - Published Version Available under License Creative Commons Attribution. Download (23kB)

Official URL: http://ipublishing.intimal.edu.my/jods.html

Abstract

This study compares four machine learning algorithms Logistic Regression, Random Forest, Support Vector Machine (SVM), and K-Nearest Neighbors (KNN) in water quality classification based on contaminant parameters. The purpose of this study is to evaluate and compare the performance of these algorithms in terms of accuracy. The methodology used includes data collection, preprocessing, and algorithm implementation with evaluation using crossvalidation techniques. The results showed that the application of the Stacking method with Gradient Boosting Meta-learner produced the highest accuracy of 96.00%, outperforming all other algorithms. In comparison, Random Forest achieved 95.75% accuracy, followed by SVM with 93.25% accuracy, and Logistic Regression and KNN each achieved 90.19% accuracy. This finding emphasizes that Stacking with Gradient Boosting provides much better performance in water quality classification compared to other models. This research provides new insights into the application of machine learning algorithms for water quality management as well as guidance for optimal algorithm selection.

Item Type:	Article
Uncontrolled Keywords:	Water Quality, Logistic Regression, Random Forest, Support Vector Machine (SVM), K-Nearest Neighbors (KNN)
Subjects:	Q Science > QA Mathematics > QA75 Electronic computers. Computer science Q Science > QA Mathematics > QA76 Computer software T Technology > T Technology (General)
Depositing User:	Unnamed user with email masilah.mansor@newinti.edu.my
Date Deposited:	26 Nov 2024 06:09
Last Modified:	26 Nov 2024 06:15
URI:	http://eprints.intimal.edu.my/id/eprint/2047

Actions (login required)

View Item