This study is based on XGBoost to classify water resources and predict their safety, and uses SHAP to explain the model.
pip install -r requirement.txt
Accuracy | Precision | Recall | F1-score |
---|---|---|---|
water-potability | 0.77 | 0.73 | 0.66 |
water-quality | 0.96 | 0.93 | 0.78 |