Research Article
Decision Tree Ensembles to Predict Coronavirus Disease 2019 Infection: A Comparative Study
Table 2
The comparative study of various ensemble methods. Different performance measures are used to study their performance.
| Ensemble | Accuracy | Precision | Recall | F1-measure | AUROC | AUPRC |
| Single decision tree | 0.845 | 0.386 | 0.275 | 0.321 | 0.609 | 0.247 | Random forest | 0.878 | 0.585 | 0.300 | 0.397 | 0.872 | 0.499 | Bagging | 0.876 | 0.568 | 0.313 | 0.403 | 0.858 | 0.506 | XGBoost | 0.865 | 0.533 | 0.575 | 0.536 | 0.863 | 0.505 | AdaBoost | 0.860 | 0.462 | 0.300 | 0.364 | 0.844 | 0.410 | Ensembles for imbalanced datasets | Balanced random forest (RUS) | 0.816 | 0.410 | 0.812 | 0.540 | 0.879 | 0.561 | SmoteBagging | 0.858 | 0.518 | 0.512 | 0.493 | 0.845 | 0.486 | RUSBagging | 0.835 | 0.432 | 0.675 | 0.517 | 0.881 | 0.516 | SmoteBoost | 0.868 | 0.544 | 0.562 | 0.530 | 0.849 | 0.514 | RUSBoost | 0.781 | 0.316 | 0.562 | 0.397 | 0.716 | 0.340 |
|
|
Bold numbers show the best performances.
|