Table 4.
Results for trained models using decision tree-based algorithms
| Algorithm |
Class 0 (Staying) |
Class 1 (Drops out) |
GM Score | F1 score | Training time (s) | ||
|---|---|---|---|---|---|---|---|
| Recall | Precision | Recall | Precision | ||||
| Decision Tree classifier | 0.7995 | 0.9959 | 0.8112 | 0.0654 | 0.8053 | 0.1210 | 70.70 |
| XGBoost | 0.9391 | 0.9967 | 0.8201 | 0.1888 | 0.8776 | 0.3069 | 284.82 |
| LightGBM | 0.9317 | 0.9970 | 0.8397 | 0.1754 | 0.8845 | 0.2902 | 67.18 |
| CatBoost | 0.9211 | 0.9972 | 0.8488 | 0.1569 | 0.8842 | 0.2645 | 407.24 |