xgboost可以用sklearn里的GridSearchCV吗?

  统计/机器学习 模型验证 Python    浏览次数:8919        分享
2

如果想对xgboost调参,可以用sklearn里的GridSearchCV吗?


 

TheTheThe   2017-10-20 20:59



   2个回答 
5

当然可以的。

from xgboost import XGBClassifier
from sklearn.model_selection import GridSearchCV
xgb_model = XGBClassifier(n_estimators=1)
parameters = {'learning_rate': [0.01, 0.02, 0.03], 'max_depth': [4, 5, 6]}
clf = GridSearchCV(xgb_model, parameters, scoring='roc_auc')
clf.fit(data, target)

上面这个例子就是对learning rate和max depth做grid search cv。

SofaSofa数据科学社区DS面试题库 DS面经

Lydia   2017-10-21 13:55

1

可以用GridSearchCV,前提是必须要用xgboost的sklearn api。

如果用xgb booster的话,就不能用了。

SofaSofa数据科学社区DS面试题库 DS面经

木子周   2017-10-22 07:22



  相关讨论

XGBClassifier和XGBRegressor里的base_score有什么用?

python里XGBClassifier怎么设置early stop?

XGBClassifier设置random_state报错

xgboost的sklearn接口有n_estimators参数,xgboost原始接口没有这个参数?

XGBoost使用hyperop调参的度量集

xgboost里怎么使用KS值?

把XGBRegressor保存到本地文件并调用

xgboost的FIT问题

xgboost中的参数min_child_weight是什么意思?

xgboost中的决策树的复杂度是怎么定义的?

  随便看看

二元分类为什么不能用MSE做为损失函数?

keras里sparse_categorical_crossentropy和categorical_crossentropy什么不同?

plt.show()之后matplotlib图像依然不展示

Data Application Lab怎么样?

如何获取pyspark DataFrame的行数和列数?