Skip to content

yuntech-bdrc/WaterQuality

Repository files navigation

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 usued to explain the model.

image

Requirement

pip install -r requirement.txt

Performance

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

Datasets

water-potability
water-quality

References

SHAP
TreeSHAP
XGBoost