A decision support machine learning tool for environmental bioremediation as water safety in India

Department of Computer Science & Engineering, Institute of Engineering and Management, Newtown, UEM, Kolkata, B3, Newtown Road, Action Area-III, West Bengal, India, 700160.

^* Corresponding author: This email address is being protected from spambots. You need JavaScript enabled to view it.

Abstract

This study presents a novel, interpretable Decision Support System that integrates Stacking Ensemble learning with Shapley Additive exPlanations to predict water potability and recommend bioremediation strategies. Preparation and validation based on Indian datasets with over 4000 samples was carried out and the proposed Stacking Ensemble (RF, XGBoost, SVM, KNN, LR) achieved the highest accuracy of 74.0%, with Area under the Curve of 0.806, being at about ten percent above other classifiers, and maintaining state-of-the-art level standards. Further beyond prediction, the framework locates significant contamination drivers (such as pH and Sulfates) and maps them to viable ecological treatments such as phytoremediation and microbial degradation, which recent studies lack. This study demonstrates a scalable, transparent AI-driven pathway for real-time water quality management in resource-constrained environments.