Research Article

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

Table 4

The performance of Standard decision tree ensembles with RUS undersampling method for different ratios of minority and majority class points. AUROC and AUPRC performance measures are used in the experiments.

EnsembleOriginalRatio = 1Ratio = 0.75Ratio = 0.50
AUROCAUPRCAUROCAUPRCAUROCAUPRCAUROCAUPRC

Single decision tree0.6590.3070.7310.2310.7330.260.7450.291
Random forest0.8720.4990.8520.4540.8590.4760.8550.454
Bagging0.8580.5060.8390.4220.8480.4450.8350.442
XGBoost0.8630.5050.8450.4750.8460.5100.8730.554
AdaBoost0.8440.4100.8260.3890.8300.4010.8490.425

Bold numbers show the best results.