WaterQuality

This study is based on XGBoost to classify water resources and predict their safety, and uses SHAP to explain the model.

Requirement

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

Name		Name	Last commit message	Last commit date
Latest commit History 5 Commits
.ipynb_checkpoints		.ipynb_checkpoints
dataset		dataset
Load_pretrain_model.ipynb		Load_pretrain_model.ipynb
README.md		README.md
X_test.csv		X_test.csv
feature_importance.csv		feature_importance.csv
model.bst		model.bst
output_df_water.csv		output_df_water.csv
output_df_water_utf8.csv		output_df_water_utf8.csv
requirement.txt		requirement.txt
water_patability_XGB_XAI.ipynb		water_patability_XGB_XAI.ipynb