From eb01acfad84bf4264e19ca39f7d447df05ded9bf Mon Sep 17 00:00:00 2001
From: hetong007 <hetong007@gmail.com>
Date: Tue, 20 Jan 2015 14:35:44 -0800
Subject: [PATCH] improve demo of cv in R

---
 R-package/R/xgb.cv.R              | 2 +-
 R-package/demo/cross_validation.R | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/R-package/R/xgb.cv.R b/R-package/R/xgb.cv.R
index e562610f1..988f67470 100644
--- a/R-package/R/xgb.cv.R
+++ b/R-package/R/xgb.cv.R
@@ -128,7 +128,7 @@ xgb.cv <- function(params=list(), data, nrounds, nfold, label = NULL, missing =
   for(line in split) dt <- line[2:length(line)] %>% str_extract_all(pattern = "\\d*\\.+\\d*") %>% unlist %>% as.list %>% {vec <- .; rbindlist(list(dt, vec), use.names = F, fill = F)}
   
   if (prediction) {
-    return(list(dt,predictValues))
+    return(list(dt = dt,pred = predictValues))
   }
   return(dt)
 }
\ No newline at end of file
diff --git a/R-package/demo/cross_validation.R b/R-package/demo/cross_validation.R
index c7e7ba537..47a0adea0 100644
--- a/R-package/demo/cross_validation.R
+++ b/R-package/demo/cross_validation.R
@@ -45,3 +45,7 @@ param <- list(max.depth=2,eta=1,silent=1)
 xgb.cv(param, dtrain, nround, nfold = 5,
        obj = logregobj, feval=evalerror)
 
+# do cross validation with prediction values for each fold
+res <- xgb.cv(param, dtrain, nround, nfold=5, prediction = TRUE)
+res$dt
+length(res$pred)