Cargando…

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

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,...

Descripción completa

Detalles Bibliográficos
Autores principales: Jain, Khushboo, Kaushik, Keshav, Gupta, Sachin Kumar, Mahajan, Shubham, Kadry, Seifedine
Formato: Online Artículo Texto
Lenguaje:English
Publicado: Nature Publishing Group UK 2023
Materias:
Acceso en línea:https://www.ncbi.nlm.nih.gov/pmc/articles/PMC10562461/
https://www.ncbi.nlm.nih.gov/pubmed/37814043
http://dx.doi.org/10.1038/s41598-023-44111-9
_version_ 1785118134022176768
author Jain, Khushboo
Kaushik, Keshav
Gupta, Sachin Kumar
Mahajan, Shubham
Kadry, Seifedine
author_facet Jain, Khushboo
Kaushik, Keshav
Gupta, Sachin Kumar
Mahajan, Shubham
Kadry, Seifedine
author_sort Jain, Khushboo
collection PubMed
description 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.
format Online
Article
Text
id pubmed-10562461
institution National Center for Biotechnology Information
language English
publishDate 2023
publisher Nature Publishing Group UK
record_format MEDLINE/PubMed
spelling pubmed-105624612023-10-11 Machine learning-based predictive modelling for the enhancement of wine quality Jain, Khushboo Kaushik, Keshav Gupta, Sachin Kumar Mahajan, Shubham Kadry, Seifedine Sci Rep Article 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. Nature Publishing Group UK 2023-10-09 /pmc/articles/PMC10562461/ /pubmed/37814043 http://dx.doi.org/10.1038/s41598-023-44111-9 Text en © The Author(s) 2023 https://creativecommons.org/licenses/by/4.0/Open Access This article is licensed under a Creative Commons Attribution 4.0 International License, which permits use, sharing, adaptation, distribution and reproduction in any medium or format, as long as you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons licence, and indicate if changes were made. The images or other third party material in this article are included in the article's Creative Commons licence, unless indicated otherwise in a credit line to the material. If material is not included in the article's Creative Commons licence and your intended use is not permitted by statutory regulation or exceeds the permitted use, you will need to obtain permission directly from the copyright holder. To view a copy of this licence, visit http://creativecommons.org/licenses/by/4.0/ (https://creativecommons.org/licenses/by/4.0/) .
spellingShingle Article
Jain, Khushboo
Kaushik, Keshav
Gupta, Sachin Kumar
Mahajan, Shubham
Kadry, Seifedine
Machine learning-based predictive modelling for the enhancement of wine quality
title Machine learning-based predictive modelling for the enhancement of wine quality
title_full Machine learning-based predictive modelling for the enhancement of wine quality
title_fullStr Machine learning-based predictive modelling for the enhancement of wine quality
title_full_unstemmed Machine learning-based predictive modelling for the enhancement of wine quality
title_short Machine learning-based predictive modelling for the enhancement of wine quality
title_sort machine learning-based predictive modelling for the enhancement of wine quality
topic Article
url https://www.ncbi.nlm.nih.gov/pmc/articles/PMC10562461/
https://www.ncbi.nlm.nih.gov/pubmed/37814043
http://dx.doi.org/10.1038/s41598-023-44111-9
work_keys_str_mv AT jainkhushboo machinelearningbasedpredictivemodellingfortheenhancementofwinequality
AT kaushikkeshav machinelearningbasedpredictivemodellingfortheenhancementofwinequality
AT guptasachinkumar machinelearningbasedpredictivemodellingfortheenhancementofwinequality
AT mahajanshubham machinelearningbasedpredictivemodellingfortheenhancementofwinequality
AT kadryseifedine machinelearningbasedpredictivemodellingfortheenhancementofwinequality