Skip to content

adolfosilva/wine_classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Wine Prediction Challenge

  • The goal is to predict if a given wine is White or Red wine.
  • The goal is to predict the quality of the wine.

Dataset: http://www3.dsi.uminho.pt/pcortez/wine/

Results:

Wine type challenge:

Using an ExtraTreesClassifier:

  • Weighted F1-score: 0.99676;
  • Balanced Accuracy: 0.99484;
  • Matthews Correlation Coefficient: 0.99129;

Wine quality challenge:

Using a Stacked classifier (ExtraTrees + KNN +LR):

  • Weighted F1-score: 0.9906;
  • Balanced Accuracy: 0.98092;
  • Matthews Correlation Coefficient: 0.98878;

Running:

$ jupyter lab notebooks

License

Distributed under the MIT license. See LICENSE for details.

About

Wine classification challenge

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published