/wine-prediction

Primary LanguageJupyter Notebook

Wine Quality Prediction

Predicting wine quality using machine learning algorithms based on various features.

Overview

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.

Features

  • Data loading and preprocessing
  • Training multiple machine learning models
  • Evaluating model performance using accuracy metrics
  • Comparing model results to identify the best-performing algorithm

Installation

To run this project locally, follow these steps:

  1. Clone the repository:

    git clone https://github.com/shcoide/wine-prediction.git
    cd wine-prediction
  2. Create and activate a virtual environment (recommended):

     virtualenv venv
     source venv/bin/activate
  3. Install dependencies:

     pip install -r requirements.txt

Running the Project

```bash
cd wine-quality-prediction
jupyter notebook notebooks/EDA.ipynb
python main.py
```