为什么xgboost中早期的.stop.round参数不起作用?

Question

我尝试在xgboost库的xgb.cv函数中使用early.stop.round参数，但是，我得到了一个错误。在我没有指定early.stop.round之后，该函数运行没有任何问题。我做错了什么?

下面是我的示例代码:

library(xgboost)
train = matrix(as.numeric(1:100),20,5)
Y = rep(c(0,1),10)
dtrain = xgb.DMatrix(train, label=Y)
#cross validation when early.stop.round =5, gives an error
CV = xgb.cv(data = dtrain, nround=200, nfold =2, metrics=list("auc"), 
            objective = "binary:logistic",early.stop.round = 5)

#cross validation when early.stop.round is not specified, works
CV = xgb.cv(data = dtrain, nround=200, nfold =2, metrics=list("auc"),
            objective = "binary:logistic")

我使用xgboost_0.4-2

Answer 1

看起来在使用度量参数和早期时出现了问题。同时停止。删除度量并尽早使用。用eval_metric="auc"代替