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

Table 6

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

Ensemble	Size
	20	50	100	200
Random forest	0.484	0.499	0.523	0.517
Bagging	0.524	0.506	0.490	0.488
XGBoost	0.494	0.505	0.507	0.511
AdaBoost	0.415	0.410	0.441	0.453
Ensembles for imbalanced datasets
Balanced random forest (RUS)	0.557	0.561	0.586	0.596
SmoteBagging	0.478	0.486	0.473	0.489
RUSBagging	0.508	0.516	0.532	0.552
SmoteBoost	0.537	0.514	0.518	0.516
RUSBoost	0.364	0.340	0.361	0.334

Bold numbers show the best performance.