Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix early stopping

Browse Source
This commit is contained in:
hetong007 2015-05-11 16:53:51 -07:00
parent 83ace55f51
commit 90096e718c
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
R-package/R/xgb.cv.R
View File

@ -196,7 +196,7 @@ xgb.cv <- function(params=list(), data, nrounds, nfold, label = NULL, missing =
bestScore = score bestScore = score
bestInd = i bestInd = i
} else { } else {
if (i-bestInd>early_stop_round) { if (i-bestInd>=early_stop_round) {
earlyStopflag = TRUE earlyStopflag = TRUE
cat('Stopping. Best iteration:',bestInd) cat('Stopping. Best iteration:',bestInd)
break break

2
R-package/R/xgb.train.R
View File

@ -187,7 +187,7 @@ xgb.train <- function(params=list(), data, nrounds, watchlist = list(),
bestScore = score bestScore = score
bestInd = i bestInd = i
} else { } else {
if (i-bestInd>early_stop_round) { if (i-bestInd>=early_stop_round) {
earlyStopflag = TRUE earlyStopflag = TRUE
cat('Stopping. Best iteration:',bestInd) cat('Stopping. Best iteration:',bestInd)
break break

4
R-package/demo/early_Stopping.R
View File

@ -34,6 +34,6 @@ print ('start training with early Stopping setting')
# training with customized objective, we can also do step by step training # training with customized objective, we can also do step by step training
# simply look at xgboost.py's implementation of train # simply look at xgboost.py's implementation of train
bst <- xgb.train(param, dtrain, num_round, watchlist, logregobj, evalerror, maximize = FALSE, bst <- xgb.train(param, dtrain, num_round, watchlist, logregobj, evalerror, maximize = FALSE,
earlyStopRound = 3) early.stop.round = 3)
bst <- xgb.cv(param, dtrain, num_round, nfold=5, obj=logregobj, feval = evalerror, bst <- xgb.cv(param, dtrain, num_round, nfold=5, obj=logregobj, feval = evalerror,
maximize = FALSE, earlyStopRound = 3) maximize = FALSE, early.stop.round = 3)