Ensemble Learning-Based Wine Quality Prediction Using Optimized Feature Selection and XGBoost

被引:0
作者
Tyagi, Sonam [1 ]
Rajput, Ishwari Singh [2 ]
Kumar, Bhawnesh [3 ]
Negi, Harendra Singh [3 ]
机构
[1] ABES Engn Coll, Dept Elect & Commun Engn, Ghaziabad 201009, Uttar Pradesh, India
[2] Graph Era Hill Univ, Dept Comp Sci & Engn, Haldwani 263139, Uttarakhand, India
[3] Graph Era Deemed Be Univ, Dept Comp Sci & Engn, Dehra Dun 248002, Uttarakhand, India
关键词
Wine quality; Feature selection; Machine learning; Classification; Prediction;
D O I
10.33889/IJMEMS.2025.10.5.077
中图分类号
T [工业技术];
学科分类号
08 ;
摘要
Recent years have seen the wine business flourish and become essential to the global economy. Due to rising demand for high-end wines, methods to consistently measure wine quality are needed. In this discipline, machine learning decision-making shows potential. High-dimensional data from several sources may impede processing and classification model performance. Feature selection increases learning and reduces computational costs by picking subsets of features and deleting irrelevant ones. This hybrid architecture uses machine learning to predict wine quality, including a feature selection method to find the most important information. Wrapper-based genetic algorithm (WGA) iteratively removes least significant features and trains a model with the remaining features until the needed number is obtained. We tested the proposed structure on two datasets consisting of 1,599 red wine samples and 4,898 white wine samples, each rated on a scale of 1-10. Additionally, the proposed technique is compared to other cutting-edge machine learning models in the same field. All categorization techniques predicted wine quality well, but WGA-XGB performed best. The study shows how feature selection improves wine quality prediction in different machine learning algorithms. The recommended strategy can be extended to different wine datasets or improved using advanced feature selection or machine learning models to improve forecast performance. Overall, the methodology is designed to be clear, relevant, and reproducible for assessing wine quality in real-world settings.
引用
收藏
页码:1621 / 1639
页数:19
相关论文
共 34 条
[11]  
Dahal KR, 2021, Open Journal of Statistics, V11, P278, DOI [10.4236/ojs.2021.112015, 10.4236/ojs.2021.112015, DOI 10.4236/OJS.2021.112015]
[12]   NSGA-II-XGB: Meta-heuristic feature selection with XGBoost framework for diabetes prediction [J].
Gupta, Aditya ;
Rajput, Ishwari Singh ;
Gunjan ;
Jain, Vibha ;
Chaurasia, Soni .
CONCURRENCY AND COMPUTATION-PRACTICE & EXPERIENCE, 2022, 34 (21)
[13]   Selection of important features and predicting wine quality using machine learning techniques [J].
Gupta, Yogesh .
6TH INTERNATIONAL CONFERENCE ON SMART COMPUTING AND COMMUNICATIONS, 2018, 125 :305-312
[14]   A hybrid genetic algorithm for feature selection wrapper based on mutual information [J].
Huang, Jinjie ;
Cai, Yunze ;
Xu, Xiaoming .
PATTERN RECOGNITION LETTERS, 2007, 28 (13) :1825-1844
[15]  
Isha N., 2024, P INT C COMM COMP SC, P1, DOI [10.1109/IC3SE62002.2024.10593326, DOI 10.1109/IC3SE62002.2024.10593326]
[16]   Analyzing of salient features and classification of wine type based on quality through various neural network and support vector machine classifiers [J].
Jana, Dipak Kumar ;
Bhunia, Prajna ;
Das Adhikary, Sirsendu ;
Mishra, Anjan .
RESULTS IN CONTROL AND OPTIMIZATION, 2023, 11
[17]  
John A., 2024, 2024 15 INT C COMP C, P1, DOI [10.1109/icccnt61001.2024.10724210, DOI 10.1109/ICCCNT61001.2024.10724210]
[18]   Network intrusion detection based on ensemble classification and feature selection method for cloud computing [J].
Krishnaveni, Sivamohan ;
Sivamohan, Sivanandam ;
Sridhar, Subramanian ;
Prabhakaran, Subramani .
CONCURRENCY AND COMPUTATION-PRACTICE & EXPERIENCE, 2022, 34 (11)
[19]   Red Wine Quality Prediction Using Machine Learning Techniques [J].
Kumar, Sunny ;
Agrawal, Kanika ;
Mandan, Nelshan .
2020 INTERNATIONAL CONFERENCE ON COMPUTER COMMUNICATION AND INFORMATICS (ICCCI - 2020), 2020, :495-500
[20]  
Laaksonen J, 1996, IEEE IJCNN, P1480, DOI 10.1109/ICNN.1996.549118