From c0c6951b7338e677f5d88fe5c85a9dcf7c612318 Mon Sep 17 00:00:00 2001 From: hetong Date: Mon, 19 Jan 2015 19:26:25 -0800 Subject: [PATCH] fix bug in format of input --- R-package/R/predict.xgb.Booster.R | 2 +- R-package/demo/predict_leaf_indices.R | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/R-package/R/predict.xgb.Booster.R b/R-package/R/predict.xgb.Booster.R index cdb3f3dc1..15af9f34d 100644 --- 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) } } diff --git a/R-package/demo/predict_leaf_indices.R b/R-package/demo/predict_leaf_indices.R index 1fc64ba4a..480578c1d 100644 --- a/R-package/demo/predict_leaf_indices.R +++ b/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)