This project combines machine learning and a Java GUI to create an application that predicts water potability based on various water quality parameters.
- water_quality_model.pkl: Pre-trained machine learning model for water quality prediction.
- scaler.pkl: Scaling logic for normalizing input data.
- water_potability.csv: Dataset containing water quality parameters like pH, Hardness, etc.
- train_model.py: Python script to train the machine learning model.
- predict.py: Python script for making batch predictions using the trained model.
- predict_water_quality.py: Python script to streamline the prediction process, possibly used for CLI.
- background.jpg: Background image used for the GUI.
- WaterQualityGUI.java: Java Swing-based GUI for inputting water parameters and receiving predictions.
- Compiled Classes:
- WaterQualityGUI.class: Compiled Java GUI classes.
- Machine Learning: Predicts water potability based on key parameters.
- Interactive GUI: User-friendly Java-based interface for input and output.
- Python Integration: Model training and predictions handled in Python, integrated with Java.
-
Python Setup:
- Install dependencies:
pip install pandas scikit-learn
- Train the model:
python train_model.py
- Test predictions:
python predict.py
- Install dependencies:
-
Java Setup:
- Compile the GUI:
javac WaterQualityGUI.java
- Run the GUI:
java WaterQualityGUI
- Compile the GUI:
-
Ensure Integration:
- Verify Python and Java paths for seamless interaction.