不能在python中使用平均绝对误差度量(mae)XGBoost

时间:2016-06-13 18:07:08

标签: python machine-learning xgboost

我从XGBoost开始并愿意将其用于回归问题。

我的评估指标是平均绝对误差(mae)。据我所知,我们可以将XGB与此指标(https://github.com/dmlc/xgboost/blob/master/doc/parameter.md#learning-task-parameters)一起使用。

我尝试使用带有参数metrics = ['mae']的XGB CV,但是我收到一条错误,指出mae metric不存在:XGBoostError: unknown evaluation metric type: mae

这是我的代码:

xgdmat = xgb.DMatrix(train, y_train)

our_params = {'eta': 0.5,
              'seed': 0,
              'subsample': 1,
              'colsample_bytree': 1, 
              'max_depth': 3,
              'min_child_weight': 3,
              'objective': 'reg:linear'} 

cv_xgb = xgb.cv(params = our_params, dtrain = xgdmat, num_boost_round = 3000, nfold = 5,
                metrics = ['mae'],
                early_stopping_rounds = 100)

有没有人已经遇到过这个问题,可以帮我解决这个问题吗?

非常感谢。

0 个答案:

没有答案