Table 1.
The result of parameter optimization.
Model | Parameter | Range | Used |
---|---|---|---|
RandomForest | n_estimators | [100, 200, 400, 500, 600] | 400 |
LightGBM | n_estimators | [100, 200, 400, 500] | 400 |
max_depth | [7, 8, 9, 10, 11] | 11 | |
num_leaves | [100, 200, 300, 400, 500] | 200 | |
ExtraTree | n_estimators | [100, 200, 400, 500, 600] | 500 |