quick fix

This commit is contained in:
tqchen 2015-01-19 10:00:28 -08:00
parent 1ea23d3390
commit 312546b99d

View File

@ -48,7 +48,7 @@ setMethod("predict", signature = "xgb.Booster",
if (predleaf) { if (predleaf) {
option <- option + 2 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) return(ret)
}) })