xgboost可以用sklearn里的GridSearchCV吗?

  统计/机器学习 模型验证 Python    浏览次数: 1424
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。

Lydia   2017-10-21 13:55

1

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

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

木子周   2017-10-22 07:22



  相关主题

python里XGBClassifier怎么设置early stop?   0回答

XGBClassifier设置random_state报错   1回答

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

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

xgboost的FIT问题   1回答

xgboost是怎么做到regularization的?   2回答

xgboost展开到二阶导数有什么好处啊   1回答

stack多个xgboost效果并不理想   3回答

xgboost怎么调参?   8回答

相比xgboost,GBDT有什么优势   2回答

为什么lightgbm比xgb快?   2回答

xgboost可以做回归预测吗?   2回答



回答问题时需要注意什么?

我们谢绝在回答前讲“生动”的故事。

我们谢绝“这么简单,你自己想”、“书上有的,你认真看”这类的回答;如果你认为对方的提问方式或者内容不妥,你可以直接忽略该问题,不用进行任何作答,甚至可以对该问题投反对票。

我们谢绝答非所问。

我们谢绝自己不会、硬要回答。

我们感激每一个用户在编写答案时的努力与付出!