Skip to content

Latest commit

 

History

History
30 lines (21 loc) · 1.1 KB

README.md

File metadata and controls

30 lines (21 loc) · 1.1 KB

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