Decision Tree Application to Classification Problems with Boosting Algorithm

Electronics

卷:10

Sanghyuk Lee¹  Long Zhao¹  Seon-Phil Jeong² 

[1] Depart of Mechatronics and Robotics, School of Advanced Technology, Xi’an Jiaotong-Liverpool University, Suzhou 215123, China;

[2] Division of Science and Technology, CST Programme BNU-HKBU United International College, Zhuhai 519085, China;

关键词: decision tree;    boosting algorithm;    receiver operating curve;    cross-validation;   

DOI  :  10.3390/electronics10161903

来源: DOAJ

【 摘 要 】

A personal credit evaluation algorithm is proposed by the design of a decision tree with a boosting algorithm, and the classification is carried out. By comparison with the conventional decision tree algorithm, it is shown that the boosting algorithm acts to speed up the processing time. The Classification and Regression Tree (CART) algorithm with the boosting algorithm showed 90.95% accuracy, slightly higher than without boosting, 90.31%. To avoid overfitting of the model on the training set due to unreasonable data set division, we consider cross-validation and illustrate the results with simulation; hypermeters of the model have been applied and the model fitting effect is verified. The proposed decision tree model is fitted optimally with the help of a confusion matrix. In this paper, relevant evaluation indicators are also introduced to evaluate the performance of the proposed model. For the comparison with the conventional methods, accuracy rate, error rate, precision, recall, etc. are also illustrated; we comprehensively evaluate the model performance based on the model accuracy after the 10-fold cross-validation. The results show that the boosting algorithm improves the performance of the model in accuracy and precision when CART is applied, but the model fitting time takes much longer, around 2 min. With the obtained result, it is verified that the performance of the decision tree model is improved under the boosting algorithm. At the same time, we test the performance of the proposed verification model with model fitting, and it could be applied to the prediction model for customers’ decisions on subscription to the fixed deposit business.

【 授权许可】

Unknown