/wine-quality-prediction

Predicting the quality of wine using machine learning and neural networks

Primary LanguagePython

Wine Quality Prediction

Neural network model to predict the quality of red and white wine using Keras and scikit-learn

Wine Type Classification

Feature Selection

Since each dataset comes with 12 features, not all are correlated with the type of wine, therefore it is important to select features that we are interested in by constructing a correlation matrix.

correlation matrix

The following features can be excluded as they have small correlation with wine type:

  • citric acid
  • pH
  • alcohol
  • quality