From 7c6cf4bad88e4c15d8408f5b0ec60f901bc86520 Mon Sep 17 00:00:00 2001 From: tqchen Date: Mon, 19 Jan 2015 09:59:33 -0800 Subject: [PATCH] quick fix --- R-package/R/predict.xgb.Booster.R | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/R-package/R/predict.xgb.Booster.R b/R-package/R/predict.xgb.Booster.R index b41f2428c..62a64f0b5 100644 --- a/R-package/R/predict.xgb.Booster.R +++ b/R-package/R/predict.xgb.Booster.R @@ -48,7 +48,7 @@ setMethod("predict", signature = "xgb.Booster", if (predleaf) { option <- option + 2 } - ret <- .Call("XGBoosterPredict_R", object, newdata, as.integer(predleaf), as.integer(ntreelimit), PACKAGE = "xgboost") + ret <- .Call("XGBoosterPredict_R", object, newdata, as.integer(option), as.integer(ntreelimit), PACKAGE = "xgboost") return(ret) })