fix bug in format of input

2015-01-19 19:26:25 -08:00
parent f295177b1d
commit c0c6951b73
2 changed files with 1 additions and 2 deletions
--- a/R-package/R/predict.xgb.Booster.R
+++ b/R-package/R/predict.xgb.Booster.R
@@ -56,7 +56,7 @@ setMethod("predict", signature = "xgb.Booster",
      if (length(ret) == len){
          ret <- matrix(ret,ncol = 1)
      } else {
-          ret <- matrix(ret, ncol = nrow(newdata))
+          ret <- matrix(ret, ncol = len)
          ret <- t(ret)
      }
  }