Skip to content

Doregon499/Wine-quality-prediction

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 

Repository files navigation

Wine-quality-prediction

Made by: Santiago Villarreal Use AI in order to predict the quality of a red wine This project will use the data set of kaggle https://www.kaggle.com/datasets/uciml/red-wine-quality-cortez-et-al-2009

imagen

This dataset has 1599 samples, with 12 labels

imagen

The project was made using Python and Colab. The code contains multiple tests on the data in order to obtain the best method for analysis. For the develop of the project I follow the next steps

  1. Import all the libraries needed
  2. Import the .cvs with all the data and clean the nan values from it
  3. Made some plots for a visual analisis
  4. Get the x and y values and impor our clean data
  5. split the data in 70% 20% and 10% for train validate and test

imagen

  1. standarization
  2. PCA

imagen

10.Train and test KNN and Decision tree with crossvalidation

Results: KNN

imagen

Decision tree

imagen

CONCLUSION: Our models obtained very low results, since there is a large number of data which, even having the same range, results in a different result, this can be seen in the visual analysis, so the data is found with very little relationship, also as It can be seen in the test and validate results, the model overfits the test data but does not generalize to the rest of the data.

Youtube link:https://youtu.be/uxIVAhF40Fc

About

Use AI in order to predict the quality of a red wine

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published