A firefly optimization algorithm for hyperparameter tuning of the support vector classifier to predict water potability

Clean water is essential for human health and life, and assessing its potability is critical for safeguarding public well-being. Machine Learning (ML) algorithms have been widely used for water potability classification based on various water quality parameters. However, the performance of these models strongly depends on effective hyperparameter tuning, which remains both challenging and resource-intensive. This study addresses this issue by proposing the Firefly Optimization Algorithm (FOA) to optimize the Support Vector Classifier (SVC) for water potability classification. Traditional hyperparameter tuning methods, such as GridSearchCV and RandomizedSearchCV, often lack the efficiency and effectiveness needed for achieving optimal model performance. In contrast, the proposed FOA-based approach provides a robust solution, demonstrating superior results compared with traditional methods. Model performance was evaluated using accuracy, precision, recall, F1-score, and Area Under the Curve (AUC). The FOA-tuned SVC achieved an accuracy of 0.6773 and an AUC of 0.7065, outperforming models tuned with conventional methods. These findings highlight the potential of nature-inspired optimization techniques, such as the Firefly algorithm, to enhance ML model performance and offer a promising approach to water potability classification.

Copyright (c) 2025 Anupkumar Bongale, Amith C. Shekhar, Santoshkumar Biradar, Kavita Tukaram Patil, Yogeshwari V. Mahajan, Deepak Dharrao, Siddhaling Urolagin, P. Olof Olsson
This work is licensed under a Creative Commons Attribution 4.0 International License.

Authors who publish with this journal agree to the following terms:

- Authors retain the copyright and grant the journal the right of first publication with the work simultaneously licensed under a Creative Commons Attribution License that allows others to share the work with an acknowledgement of the work's authorship and initial publication in this journal.

- Authors are able to enter into separate, additional contractual arrangements for the non-exclusive distribution of the journal's published version of the work (e.g., post it to an institutional repository or publish it in a book), with an acknowledgement of its initial publication in this journal.

- Authors are permitted and encouraged to post their work online (e.g., in institutional repositories or on their website) after its publication in ETASR with an acknowledgement of its initial publication in this journal.