Table 4 Comparison of the performance of XGBoost and RF models by confusion matrix and evaluation measures.
Models | Testing dataset | ||||||
---|---|---|---|---|---|---|---|
Predict ineffective | Predict effective | Accuracy | Precision | Recall | F1-score | AUC | |
Extreme Gradient Boosting (XGBoost) | |||||||
ineffective | 4 | 1 | 0.8750 | 0.8750 | 0.8750 | 0.8750 | 0.8636 |
effective | 1 | 10 | |||||
Random Forest (RF) | |||||||
ineffective | 4 | 1 | 0.8125 | 0.8271 | 0.8125 | 0.8166 | 0.8818 |
effective | 2 | 9 |