XGBoost模型参数

参数含义

参数列表

参数取值

n_estimators

迭代次数

[10, 20, 30, …, 190, 200]

180

max_depth

树的最大深度

[10, 11, …, 19, 20]

14

min_child_weight

子节点中最小的样本权重和

[0, 1, 2, 3, 4, 5]

4

subsample

训练模型的子样本占整个样本集合的比例

[0.6, 0.7, 0.8, 0.9, 1]

0.9

colsample_bytree

构建树时对特征随机采样的比例

[0.6, 0.7, 0.8, 0.9, 1]

0.9

reg_alpha

L1正则化系数

[1, 2, …, 9, 10]

1

reg_lambda

L2正则化系数

[1, 2, …, 9, 10]

8

learning_rate

学习率

[0, 0.1, …, 0.9, 1]

0.6