fix the reshape and predleaf arguments (#244)

8da60f68 · Yachen Yan · Guolin Ke · 112ab6f9 · 8da60f68
Commit 8da60f68 authored Jan 21, 2017 by Yachen Yan Committed by Guolin Ke Jan 22, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

R-package/R/lgb.Predictor.R R-package/R/lgb.Predictor.R +5 -1

No files found.
--- a/R-package/R/lgb.Predictor.R
+++ b/R-package/R/lgb.Predictor.R
@@ -85,7 +85,11 @@ Predictor <- R6Class(
        stop("predict: prediction length ", sQuote(length(preds))," is not a multiple of nrows(data): ", sQuote(num_row))
      }
      npred_per_case <- length(preds) / num_row
-      if (reshape && npred_per_case > 1) { preds <- matrix(preds, ncol = npred_per_case) }
+      if (predleaf) {
+        preds <- matrix(preds, ncol = npred_per_case, byrow = TRUE)
+      } else if (reshape && npred_per_case > 1) {
+        preds <- matrix(preds, ncol = npred_per_case, byrow = TRUE)
+      }
      preds
    }
  ),