Table A1.

Hyperparameter grid

HyperparameterDescriptionSearch spaceModel
num_iterationsNumber of boosting iterationsFixed to 100 by the developers to
Reduce down the search space to
Facilitate efficient optimization
Learning_rate hyperparameter is
Tuned as a proxy for the effect of
Lesser or higher number of trees
100
feature_fractionLightGBM will randomly select
A subset of features on each
Iteration (tree), if
feature fraction is smaller than 1.0. For example, if you set it to
0.8, LightGBM will select 80%
Of features before training each
tree
• can be used to speed up training
• can be used to deal with over-fitting
0.3, 0.5, 0.7, 0.90.3
max_depth*
* Important
hyperparameter
Limits the max depth of a tree
This is used to deal with overfitting when #data is small. Tree
Still grows leaf-wise
2, 4, 8, 104
learning_rate*
* Important
hyperparameter
A technique to slow down the
Learning by weighting the
Corrections by new trees at
Every learning iteration
0.001, 0.01, 0.05, 0.10.1
min_gain_to_splitMinimal gain to perform0.3, 0.5, 0.70.3
min_data_in_leafMinimal number of data in one
Leaf. Can be used to deal with
over-fitting
5, 10, 15, 2015
lambda_l1L1 regularization0.1, 0.30.3
scale_pos_weightWeight of labels with positive
class
1.0, 1.5, 1.71.7

Source(s): Authors’ own creation/work

or Create an Account

Close subscription notice
Close access options