我正在尝试在 XGBRegressor 上运行 GRIDSEARCHCV (sklearn)。关于参数的文档说,如果回归,那么objective = reg:squarederror。(请参阅https://github.com/dmlc/xgboost/tree/master/demo/regression)但是,每当我尝试运行搜索时,我都会收到一条错误消息XGBoostError: b'[13:39:54] src/objective/objective.cc:23: Unknown objective function reg:squarederror.
我不知道如何解决这个问题。为了完整起见,下面是我为此目的使用的一段代码。
cv_params = {
'n_estimators' : np.arange(100, 1201, 100),
'max_depth' : np.arange(2, 10)
}
xgbr_params = {'objective':'reg:squarederror','n_jobs':-1,'random_state':4444,'min_child_weight':1,
'eta':0.3,'subsample':0.8,'gamma':0.5,'colsample_bytree':0.8}
opt_xgbr = GridSearchCV(xgb.XGBRegressor(**xgbr_params)
,param_grid=cv_params,scoring='r2',cv=5,n_jobs=-1,return_train_score=True, verbose=3)
任何帮助将不胜感激。
谢谢