Table 2 Range of hyper-parameter tuning for chosen ML models.
Model | Hyper parameters range | ||
|---|---|---|---|
n_estimators | learning_rate | max_depth | |
XGBoost | 50–1000 | 0.01–0.3 | 3–15 |
AdaBoost | 50–500 | 0.01–1.0 | Not used (base estimator controls depth) |
GBM | 50–1000 | 0.01–0.3 | 3–10 |
CatBoost | 100–1000 | 0.01–0.3 | 4–10 |