我从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)
有没有人已经遇到过这个问题,可以帮我解决这个问题吗?
非常感谢。