Table 1 Hyperparameter values used in the grid search for the tuning of the XGBoost models.
Hyperarameter | Set of values |
---|---|
eta | [0.001, 0.01, 0.1, 0.2, 0.3] |
gamma | [0.05, 0.5, 1, 1.5] |
min_child_weight | [5, 7, 9, 10] |
subsample | [0.5, 0.8, 1] |
colsample_bytree | [0.6, 0.8, 1] |
lambda | [0.1, 0.5, 1] |
alpha | [0, 1, 2] |