From 431277d5ca12973c506f3288df1d0febde981685 Mon Sep 17 00:00:00 2001
From: unknown <hetong007@gmail.com>
Date: Sun, 29 Mar 2015 00:02:29 -0700
Subject: [PATCH] fix multi cv pred

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

diff --git a/R-package/R/xgb.cv.R b/R-package/R/xgb.cv.R
index 8d17c2d8f..3472ead4d 100644
--- a/R-package/R/xgb.cv.R
+++ b/R-package/R/xgb.cv.R
@@ -121,7 +121,8 @@ xgb.cv <- function(params=list(), data, nrounds, nfold, label = NULL, missing =
         } else {
           res <- xgb.iter.eval(fd$booster, fd$watchlist, i - 1, feval, prediction)
           if (mat_pred) {
-            predictValues[fd$index,] <- res[[2]]
+            pred_mat = matrix(res[[2]],num_class,length(fd$index))
+            predictValues[fd$index,] <- t(pred_mat)
           } else {
             predictValues[fd$index] <- res[[2]]
           }