Research Article

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

Table 5

The performance (AUROC) of various ensembles of decision trees for different sizes of ensembles.

EnsembleSize
2050100200

Random forest0.8520.8720.8820.876
Bagging0.8590.8580.8560.857
XGBoost0.8630.8640.8670.868
AdaBoost0.8270.8440.8500.855
Ensembles for imbalanced datasets
Balanced random Forest (RUS)0.8710.8790.8890.893
SmoteBagging0.8570.8460.8540.859
RUSBagging0.8740.8810.8800.883
SmoteBoost0.8530.8490.8640.864
RUSBoost0.75160.7160.7170.715

Bold numbers show the best performance.