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:
commit
c44eb3ab91
@ -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)) {
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user