Skip to content

Commit

Permalink
[R] fix warning at prediction (#1323)
Browse files Browse the repository at this point in the history
* Update lgb.Booster.R

* Update lgb.Booster.R
  • Loading branch information
guolinke committed Apr 19, 2018
1 parent ebf962f commit 1e3b57d
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion R-package/R/lgb.Booster.R
Expand Up @@ -410,6 +410,7 @@ Booster <- R6Class(
num_iteration = NULL,
rawscore = FALSE,
predleaf = FALSE,
predcontrib = FALSE,
header = FALSE,
reshape = FALSE, ...) {

Expand All @@ -420,7 +421,7 @@ Booster <- R6Class(

# Predict on new data
predictor <- Predictor$new(private$handle, ...)
predictor$predict(data, num_iteration, rawscore, predleaf, header, reshape)
predictor$predict(data, num_iteration, rawscore, predleaf, predcontrib, header, reshape)

},

Expand Down

0 comments on commit 1e3b57d

Please sign in to comment.