{"title":"基于机器学习算法的葡萄酒质量预测模型构建","authors":"Haoyu Zhang, Zhile Wang, Jiawei He, Jijiao Tong","doi":"10.1145/3480433.3480443","DOIUrl":null,"url":null,"abstract":"In the study, our group choose a set of quality of red wine as data set. To get a more accurate result, we turn the quality into binary classification. And we try to build models to predict the quality of red wine based on machine learning algorithms, including Decision Tree, Boosting, Classification and regression tree and Random Forest. Among them, CART and Random Forest both get a high accuracy. A binary tree is built with CART and feature importance is analyzed. Meanwhile, we try to combine logistic algorithm with Random Forest and compare the accuracy of different models. In this way, it's found that there is a way to improve the accuracy of these models.","PeriodicalId":415865,"journal":{"name":"2021 5th International Conference on Artificial Intelligence and Virtual Reality (AIVR)","volume":"345 1","pages":"0"},"PeriodicalIF":0.0000,"publicationDate":"2021-07-23","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":"1","resultStr":"{\"title\":\"Construction of Wine Quality Prediction Model based on Machine Learning Algorithm\",\"authors\":\"Haoyu Zhang, Zhile Wang, Jiawei He, Jijiao Tong\",\"doi\":\"10.1145/3480433.3480443\",\"DOIUrl\":null,\"url\":null,\"abstract\":\"In the study, our group choose a set of quality of red wine as data set. To get a more accurate result, we turn the quality into binary classification. And we try to build models to predict the quality of red wine based on machine learning algorithms, including Decision Tree, Boosting, Classification and regression tree and Random Forest. Among them, CART and Random Forest both get a high accuracy. A binary tree is built with CART and feature importance is analyzed. Meanwhile, we try to combine logistic algorithm with Random Forest and compare the accuracy of different models. In this way, it's found that there is a way to improve the accuracy of these models.\",\"PeriodicalId\":415865,\"journal\":{\"name\":\"2021 5th International Conference on Artificial Intelligence and Virtual Reality (AIVR)\",\"volume\":\"345 1\",\"pages\":\"0\"},\"PeriodicalIF\":0.0000,\"publicationDate\":\"2021-07-23\",\"publicationTypes\":\"Journal Article\",\"fieldsOfStudy\":null,\"isOpenAccess\":false,\"openAccessPdf\":\"\",\"citationCount\":\"1\",\"resultStr\":null,\"platform\":\"Semanticscholar\",\"paperid\":null,\"PeriodicalName\":\"2021 5th International Conference on Artificial Intelligence and Virtual Reality (AIVR)\",\"FirstCategoryId\":\"1085\",\"ListUrlMain\":\"https://doi.org/10.1145/3480433.3480443\",\"RegionNum\":0,\"RegionCategory\":null,\"ArticlePicture\":[],\"TitleCN\":null,\"AbstractTextCN\":null,\"PMCID\":null,\"EPubDate\":\"\",\"PubModel\":\"\",\"JCR\":\"\",\"JCRName\":\"\",\"Score\":null,\"Total\":0}","platform":"Semanticscholar","paperid":null,"PeriodicalName":"2021 5th International Conference on Artificial Intelligence and Virtual Reality (AIVR)","FirstCategoryId":"1085","ListUrlMain":"https://doi.org/10.1145/3480433.3480443","RegionNum":0,"RegionCategory":null,"ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":null,"EPubDate":"","PubModel":"","JCR":"","JCRName":"","Score":null,"Total":0}
Construction of Wine Quality Prediction Model based on Machine Learning Algorithm
In the study, our group choose a set of quality of red wine as data set. To get a more accurate result, we turn the quality into binary classification. And we try to build models to predict the quality of red wine based on machine learning algorithms, including Decision Tree, Boosting, Classification and regression tree and Random Forest. Among them, CART and Random Forest both get a high accuracy. A binary tree is built with CART and feature importance is analyzed. Meanwhile, we try to combine logistic algorithm with Random Forest and compare the accuracy of different models. In this way, it's found that there is a way to improve the accuracy of these models.