Table 4 Key hyperparameters for ML models.
ML model | Hyperparameters |
---|---|
RF | max_depth = 5 |
max_features = 7 | |
n_estimators = 115 | |
XGBoost | Colsample_bytree = 0.41 |
learning_rate = 0.28 | |
max_depth = 2 | |
n_estimators = 260 | |
LightGBM | bagging_fraction = 0.71 |
feature_fraction = 0.73 | |
learning_rate = 0.32 | |
max_depth = 13 | |
num_leaves = 290 |