WineQualityPrediction

Given a set of features as inputs, the task here is to predict the quality of wine on a scale of 0 - 10. I have solved it as a regression problem using Linear Regression.

Dataset

The dataset used here is Wine Quality Data set from UCI Machine Learning Repository. I have attached the csv file needed for the regression task - winequality-red.csv The same can also be found here https://archive.ics.uci.edu/ml/datasets/Wine+Quality

Input variables (based on physicochemical tests):

  1. fixed acidity
  2. volatile acidity
  3. citric acid
  4. residual sugar
  5. chlorides
  6. free sulfur dioxide
  7. total sulfur dioxide
  8. density
  9. pH
  10. sulphates
  11. alcohol

Output variable (based on sensory data): quality (score between 0 and 10)

Dependencies

  1. Python
  2. Pandas
  3. matplotlib
  4. numpy
  5. scikit-learn

Setup

  1. Clone the repository
  2. pip install -r reqirements.txt
  3. Run the main.py to see the results.

Output

Correlation between different attributes

1

RMSE

2

Coefficient values

3