From 48e19c196401c2774012768ab0d64ee3de55e137 Mon Sep 17 00:00:00 2001
From: Tong He <hetong007@gmail.com>
Date: Mon, 22 Jun 2015 12:42:12 -0700
Subject: [PATCH] Update xgb.cv.R

---
 R-package/R/xgb.cv.R | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/R-package/R/xgb.cv.R b/R-package/R/xgb.cv.R
index 06e2cf82e..793d904cd 100644
--- a/R-package/R/xgb.cv.R
+++ b/R-package/R/xgb.cv.R
@@ -216,7 +216,11 @@ xgb.cv <- function(params=list(), data, nrounds, nfold, label = NULL, missing =
     if (prediction) {
         for (k in 1:nfold) {
             fd = xgb_folds[[k]]
-            res = xgb.iter.eval(fd$booster, fd$watchlist, i - 1, feval, prediction)
+            if (!is.null(early.stop.round) && earlyStopflag) {
+              res = xgb.iter.eval(fd$booster, fd$watchlist, bestInd - 1, feval, prediction)
+            } else {
+              res = xgb.iter.eval(fd$booster, fd$watchlist, nrounds - 1, feval, prediction)
+            }
             if (mat_pred) {
                 pred_mat = matrix(res[[2]],num_class,length(fd$index))
                 predictValues[fd$index,] = t(pred_mat)