Implementation of GridSearchCV to Find the Best Hyperparameter Combination for Classification Model Algorithm in Predicting Water Potability

Drinking water quality is an important factor in public health, so an accurate approach is needed to determine water potability. This research aims to create a water potability prediction model using machine learning methods, with a focus on model accuracy and testing. The dataset used includes various chemical parameters, as well as one radiological and acceptability parameter. In this study, various machine learning algorithms, such as Support Vector Machine (SVM), Random Forest (RF), and Logistic Regression, were applied using GridSearchCV and their performance compared. Models were evaluated using accuracy, precision, recall, F1-score, and confusion matrix metrics, with cross-validation to ensure generalizability. The results showed that the Support Vector Machine algorithm provided the best performance with an accuracy of 70.43%, followed by Random Forest and Logistic Regression with accuracies of 70.12% and 62.20%, respectively. The Support Vector Machine-based model is able to provide reliable predictions and can be used as a tool to support decision-making in water quality management.

Authors

Institutions

Publication Details

Journal
Journal of Artificial Intelligence and Engineering Applications (JAIEA)
Published
2025-02-15
DOI
https://doi.org/10.59934/jaiea.v4i2.844
Citations
2
Primary Topic
Water Quality Monitoring Technologies
Type
article
Field-Weighted Citation Impact
1.63
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Implementation of GridSearchCV to Find the Best Hyperparameter Combination for Classification Model Algorithm in Predicting Water Potability

Aliyah Kurniasih, Cantika Previana
2 citations
Journal of Artificial Intelligence and Engineering Applications (JAIEA)
Water Quality Monitoring Technologies
1.63
article

Implementation of GridSearchCV to Find the Best Hyperparameter Combination for Classification Model Algorithm in Predicting Water Potability

Aliyah Kurniasih, Cantika Previana
article en
2 citations

Abstract

Drinking water quality is an important factor in public health, so an accurate approach is needed to determine water potability. This research aims to create a water potability prediction model using machine learning methods, with a focus on model accuracy and testing. The dataset used includes various chemical parameters, as well as one radiological and acceptability parameter. In this study, various machine learning algorithms, such as Support Vector Machine (SVM), Random Forest (RF), and Logistic Regression, were applied using GridSearchCV and their performance compared. Models were evaluated using accuracy, precision, recall, F1-score, and confusion matrix metrics, with cross-validation to ensure generalizability. The results showed that the Support Vector Machine algorithm provided the best performance with an accuracy of 70.43%, followed by Random Forest and Logistic Regression with accuracies of 70.12% and 62.20%, respectively. The Support Vector Machine-based model is able to provide reliable predictions and can be used as a tool to support decision-making in water quality management.

Journal of Artificial Intelligence and Engineering Applications (JAIEA)Vol. 4(2)
ARS University
Openalex Percentile: Top 19%
Water Quality Monitoring Technologies
1.63
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.