WaterQuality

This study aims to classify water resources, predict their safety, and explain the trained model. XGBoost is used to construct the model and predict the water datasets. SHAP is used to explain the model.

Requirement

pip install -r requirement.txt

Performance

Dataset	Accuracy	Precision	Recall	F1-score
water-potability	0.77	0.73	0.66	0.7
water-quality	0.96	0.93	0.78	0.85

Datasets

water-potability
water-quality

References

SHAP
TreeSHAP
XGBoost