XGBoost 预测都一样

数据挖掘 xgboost
2022-02-26 04:45:33

当我评估模型时,我似乎得到了不错的 RMSE 分数,但是当我尝试在调用模型时实际查看预测时,我的所有值都是相同的。

 xdata = xgboost.DMatrix(X_train, y_train, feature_names=all_vars)
 xdata_val = xgboost.DMatrix(X_valid, y_valid, feature_names=all_vars)
 xgb_parms['seed'] = random.randint(0,1e9)
 model = xgboost.train(xgb_parms, xdata)
 model.eval(xdata_val)
 ypred = model.predict(xdata_val)

我相信错误出现在我的最后一步,我做错了什么?

1个回答

确保向模型传递“objective”参数,并为“eval_metric”参数使用“rmse”。

https://github.com/dmlc/xgboost/blob/master/doc/parameter.rst