Skip to content

An attempt to study various ML models for predicting the quality of Red Wine using various performance measures.

Notifications You must be signed in to change notification settings

anuj-rai-23/Model-Ananlysis-and-Red-Wine-Quality-Prediction

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 

Repository files navigation

Model-Ananlysis-and-Red-Wine-Quality-Prediction

An attempt to study various ML models for predicting the quality of Red Wine using various performance measures.
A project for Machine Learning course (CS-503) done under supervision of Dr. CK Narayanan

Dataset

Kaggle Red Wine Quality Dataset is heavily biased towards 3 classes. Converted it into a two class dataset and performed binary classification.

ML Models used

  • Decision Tree
  • Random Forests
  • Random Gradient Boost
  • Support Vector Machine

Performance Measures

  • Train Accuracy
  • Test Accuracy
  • F1-Score
  • Precision Score
  • Recall

For all details of feature selection, model analysis and results see notebook