Telematika : Jurnal Informatika dan Teknologi Informasi
Vol 21 No 2 (2024): Edisi Juni 2024

Performance Evaluation of Multiple Deep Learning Models for Wine Quality Prediction

Fabiyanto, Dedik (Unknown)
Rianto, Yan (Unknown)



Article Info

Publish Date
20 Jun 2024

Abstract

Research utilizing a dataset from the UCI repository evaluated the predictive accuracy of nine machine learning models for wine quality. The models employed include Logistic Regression, K-Nearest Neighbor (KNN), Decision Tree, Support Vector Machine (SVM), Random Forest, XGBoost, LightGBM, CatBoost, and Gradient Boosting. The dataset comprises 1,599 samples with 12 chemical parameters. Data preprocessing, including oversampling, normalization, standardization, and seeding, was performed to enhance model performance.The study's findings indicate that the models with the highest accuracy values were LightGBM (87.80%), CatBoost (86.60%), and Random Forest (85.70%). A voting classifier combining these three models achieved an accuracy of 87.29%. Further analysis using a confusion matrix demonstrated that this combined model effectively predicts the "Good" and "Not Good" classes.In conclusion, the combination of LightGBM, CatBoost, and Random Forest models proves to be an effective approach for predicting wine quality based on chemical parameters, with an accuracy value of 87.29%.

Copyrights © 2024