BEST: a decision tree algorithm that handles missing values

被引:21
作者
Beaulac, Cedric [1 ]
Rosenthal, Jeffrey S. [1 ]
机构
[1] Univ Toronto, Dept Stat Sci, Toronto, ON, Canada
基金
加拿大自然科学与工程研究理事会;
关键词
Classification and regression tree; Missing data; Applied machine learning; Interpretable models; Variable importance analysis; CLASSIFICATION TREES; IMPUTATION; INFERENCE;
D O I
10.1007/s00180-020-00987-z
中图分类号
O21 [概率论与数理统计]; C8 [统计学];
学科分类号
020208 ; 070103 ; 0714 ;
摘要
The main contribution of this paper is the development of a new decision tree algorithm. The proposed approach allows users to guide the algorithm through the data partitioning process. We believe this feature has many applications but in this paper we demonstrate how to utilize this algorithm to analyse data sets containing missing values. We tested our algorithm against simulated data sets with various missing data structures and a real data set. The results demonstrate that this new classification procedure efficiently handles missing values and produces results that are slightly more accurate and more interpretable than most common procedures without any imputations or pre-processing.
引用
收藏
页码:1001 / 1026
页数:26
相关论文
共 30 条
[1]   Grades and incentives: assessing competing grade point average measures and postgraduate outcomes [J].
Bailey, Michael A. ;
Rosenthal, Jeffrey S. ;
Yoon, Albert H. .
STUDIES IN HIGHER EDUCATION, 2016, 41 (09) :1548-1562
[2]  
BEAULAC C, 2018, PREDICTING U STUDENT
[3]   Random forests [J].
Breiman, L .
MACHINE LEARNING, 2001, 45 (01) :5-32
[4]  
Breiman L., 1984, Classification and Regression Trees, DOI DOI 10.1201/9781315139470
[5]  
Ding YF, 2010, J MACH LEARN RES, V11, P131
[6]  
FEELDERS AJ, 1999, PKDD
[7]  
FRIEDMAN J, 1997, LAZY DECISION TREES
[8]   Decision Tree: Review of Techniques for Missing Values at Training, Testing and Compatibility [J].
Gavankar, Sachin ;
Sawarkar, Sudhirkumar .
2015 THIRD INTERNATIONAL CONFERENCE ON ARTIFICIAL INTELLIGENCE, MODELLING AND SIMULATION (AIMS 2015), 2015, :122-126
[9]   Extremely randomized trees [J].
Geurts, P ;
Ernst, D ;
Wehenkel, L .
MACHINE LEARNING, 2006, 63 (01) :3-42
[10]  
Hastie T., 2009, The Elements of Statistical Learning: Data Mining, Inference, and Prediction, V2nd ed., DOI DOI 10.1007/B94608