Research Article

Decision Tree Ensembles to Predict Coronavirus Disease 2019 Infection: A Comparative Study

Table 7

The effect of age variable on the performance (AUROC) of various ensembles. “+” sign shows the better performance of with age or without age.

EnsembleSize = 50Size = 100
With ageWithout ageWith ageWithout age

Random forest0.8670.872 (+)0.8780.882 (+)
Bagging0.861 (+)0.8580.861 (+)0.856
XGBoost0.8600.863 (+)0.8660.867 (+)
AdaBoost0.850 (+)0.8440.865 (+)0.850
Ensembles for imbalanced datasets
Balanced random Forest (RUS)0.892 (+)0.8790.895 (+)0.889
SmoteBagging0.862 (+)0.8450.859 (+)0.854
RUSBagging0.883 (+)0.8810.887 (+)0.880
SmoteBoost0.863 (+)0.8490.8570.863 (+)
RUSBoost0.755 (+)0.7160.727 (+)0.7171