Predicting wine quality using machine learning algorithms based on various features.
This project aims to predict the quality of wine based on its chemical properties using several machine learning algorithms. It includes data preprocessing, model training, evaluation, and comparison to determine the best-performing model for predicting wine quality.
- Data loading and preprocessing
- Training multiple machine learning models
- Evaluating model performance using accuracy metrics
- Comparing model results to identify the best-performing algorithm
To run this project locally, follow these steps:
-
Clone the repository:
git clone https://github.com/shcoide/wine-prediction.git cd wine-prediction
-
Create and activate a virtual environment (recommended):
virtualenv venv source venv/bin/activate
-
Install dependencies:
pip install -r requirements.txt
```bash
cd wine-quality-prediction
jupyter notebook notebooks/EDA.ipynb
python main.py
```