Table 1 Hyperparameter space searched in GridSearchCV for the XGBoost model after feature selection
Hyperparameter | Values searched |
|---|---|
Learning rate | .001, .005, .01, .05, .10, .15 |
Max depth | 5, 7, 9, 12, 15 |
No. Estimators | 50, 100, 200, 500, 700 |
Min. child weight | 1, 3, 5, 7 |
Gamma | .0, .1, .2 |