Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[R] fix #1903 (#1929)

Browse Source
This commit is contained in:
Vadim Khotilovich
2017-01-06 13:16:37 -08:00
committed by Tianqi Chen
parent d7406e07f3
commit 87e897f428
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
R-package/R/xgb.Booster.R
+2
View File
@@ -185,6 +185,8 @@ predict.xgb.Booster <- function(object, newdata, missing = NA,
newdata <- xgb.DMatrix(newdata, missing = missing) newdata <- xgb.DMatrix(newdata, missing = missing)
if (is.null(ntreelimit)) if (is.null(ntreelimit))
ntreelimit <- NVL(object$best_ntreelimit, 0) ntreelimit <- NVL(object$best_ntreelimit, 0)
if (NVL(object$params[['booster']], '') == 'gblinear')
ntreelimit <- 0
if (ntreelimit < 0) if (ntreelimit < 0)
stop("ntreelimit cannot be negative") stop("ntreelimit cannot be negative")