Table 5 Model parameter settings.

From: Hydropower station scheduling with ship arrival prediction and energy storage

Parameter

Value

Description

n_estimators

500

Maximum number of iterations of the weak learner

learning_rate

0.05

Control the iteration rate to prevent over-fitting

gamma

0

The minimum descent value of the loss function required for node splitting, the larger the parameter value, the more conservative the algorithm

\(max\_depth\)

9

The maximum depth of the tree, the larger the value the more complex the model. Overfitting can be controlled by this value

alpha

1

L1 regularization term for the weights

lambda

0.5

L2 regularization term for the weights

subsample

1

Control the proportion of random sampling for each tree

booster

gbtree

Select base classifier, specify ascent model, commonly tree or linear model