如何解决GridSearchCV:XGBRegressor 中的参数错误
我使用 GridSearchCV 在我的 XGB 模型中找到最佳参数,这是我的代码
# Xgboost
grid_xgb = {
'booster': 'gbtree','objective': 'binary:logistic','subsample': [0.6,0.7,0.8,0.9],'colsample_bytree': [0.6,'eta': [0.05,0.1,0.2,0.3],'max_depth': [3,5,7],'seed': [2021,2022],'eval_metric': 'logloss'
}
model_xgb = xgb.XGBRegressor()
search_xgb = GridSearchCV(estimator = model_xgb,param_grid = grid_xgb,cv = 5,n_jobs = -1,verbose = 2)
但是我得到一个错误信息:
Parameter values for parameter (booster) need to be a sequence(but not a string) or np.ndarray
我检查了 XGB 文档,上面写着
"助推器 [默认= gbtree ]
使用哪个助推器。可以是 gbtree、gblinear 或 dart; gbtree 和 dart 使用基于树的模型,而 gblinear 使用线性函数。"
所以,我不知道为什么会发生错误以及如何修复它。
解决方法
如 scikit-learn documentation 中所述,不同的参数值需要作为列表传递给 GridSearchCV
,这意味着 booster
、objective
和 {{ 1}} 也需要用方括号括起来:
eval_metric
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。