fix bug in format of input

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)
       }
   }

R-package/demo/predict_leaf_indices.R

@@ -12,7 +12,6 @@ nround = 5
 # training the model for two rounds
 bst = xgb.train(param, dtrain, nround, watchlist)
 cat('start testing prediction from first n trees\n')
-labels <- getinfo(dtest,'label')
 
 ### predict using first 2 tree
 pred_with_leaf = predict(bst, dtest, ntreelimit = 2, predleaf = TRUE)