Merge pull request #730 from ganesh-krishnan/master

Fixed off by 1 bug in early.stop.rounds in xgb.cv
This commit is contained in:
Yuan (Terry) Tang 2016-01-10 13:14:50 -05:00
commit c44eb3ab91

View File

@ -191,7 +191,7 @@ xgb.cv <- function(params=list(), data, nrounds, nfold, label = NULL, missing =
# early_Stopping
if (!is.null(early.stop.round)){
score <- strsplit(ret,'\\s+')[[1]][1 + length(metrics) + 2]
score <- strsplit(ret,'\\s+')[[1]][2 + length(metrics)]
score <- strsplit(score,'\\+|:')[[1]][[2]]
score <- as.numeric(score)
if ( (maximize && score > bestScore) || (!maximize && score < bestScore)) {