Back to Search Start Over

Machine learning-based predictive modelling for the enhancement of wine quality.

Authors :
Jain, Khushboo
Kaushik, Keshav
Gupta, Sachin Kumar
Mahajan, Shubham
Kadry, Seifedine
Source :
Scientific Reports. 10/9/2023, Vol. 13 Issue 1, p1-18. 18p.
Publication Year :
2023

Abstract

The certification of wine quality is essential to the wine industry. The main goal of this work is to develop a machine learning model to forecast wine quality using the dataset. We utilised samples from the red wine dataset (RWD) with eleven distinct physiochemical properties. With the initial RWD, five machine learning (ML) models were trained and put to the test. The most accurate algorithms are Random Forest (RF) and Extreme Gradient Boosting (XGBoost). Using these two ML approaches, the top three features from a total of eleven features are chosen, and ML analysis is performed on the remaining features. Several graphs are employed to demonstrate the feature importance based on the XGBoost model and RF. Wine quality was predicted using relevant characteristics, often referred to as fundamental elements, that were shown to be essential during the feature selection procedure. When trained and tested without feature selection, with feature selection (RF), and with key attributes, the XGBoost classifier displayed 100% accuracy. In the presence of essential variables, the RF classifier performed better. Finally, to assess the precision of their predictions, the authors trained an RF classifier, validated it, and changed its hyperparameters. To address collinearity and decrease the quantity of predictors without sacrificing model accuracy, we have also used cluster analysis. [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
20452322
Volume :
13
Issue :
1
Database :
Academic Search Index
Journal :
Scientific Reports
Publication Type :
Academic Journal
Accession number :
172866865
Full Text :
https://doi.org/10.1038/s41598-023-44111-9