This project predicts the potability of Water using machine learning models.
After experimenting with multiple algorithms (Logistic Regression, Random Forest, Gradient Boosting, KNN, etc.), the final selected model was chosen based on accuracy, precision/recall, and overall robustness.
- Source: Kaggle - Water Quality Dataset
- Features include:
- pH
- Hardness
- Solids
- Chloramines
- Sulfate
- Conductivity
- Organic Carbon
- Trihalomethanes
- Turbidity
- Target: Potability (0 = Not Potable, 1 = Potable)
-
Data Preprocessing
- Handling missing values
- Normalization/standardization
- Train-test split
-
Model Selection
- Tested multiple models:
- Logistic Regression With Plolynominal Feature Engineering
- Logistic Regression Without Plolynominal Feature Engineering
- Random Forest
- Gradient Boosting
- KNN
- Compared performance using metrics: Accuracy, Precision, Recall, F1-score
- Tested multiple models:
-
Final Model
- Selected model: [Logistic Regression With Plolynominal Feature Engineering]
- Achieved best balance between accuracy and generalization.
Colab link: Colab Test Phase - Water Quality Dataset