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

fix with new predict

Browse Source
This commit is contained in:
hetong007 2015-02-09 16:25:00 -08:00
parent f7c838ffaa
commit 0aef62dabc
4 changed files with 24 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
R-package/R/predict.xgb.Booster.R
View File

@ -77,4 +77,4 @@ setMethod("predict", signature = "xgb.Booster",
}
return(ret)
})

16
R-package/R/predict.xgb.Booster.handle.R Normal file
View File

@ -0,0 +1,16 @@
setClass("xgb.Booster.handle")
setMethod("predict", signature = "xgb.Booster.handle",
definition = function(object, ...) {
if (class(object) != "xgb.Booster.handle"){
stop("predict: model in prediction must be of class xgb.Booster.handle")
}
bst <- list(handle = object,raw = NULL)
class(bst) <- 'xgb.Booster'
bst$raw <- xgb.save.raw(bst$handle)
ret = predict(bst, ...)
return(ret)
})

26
R-package/R/utils.R
View File

@ -102,42 +102,28 @@ xgb.numrow <- function(dmat) {
}
# iteratively update booster with customized statistics
xgb.iter.boost <- function(booster, dtrain, gpair) {
if (class(booster) != "xgb.Booster") {
stop("xgb.iter.update: first argument must be type xgb.Booster")
} else {
if (is.null(booster$handle)) {
booster$handle <- xgb.load(booster$raw)
} else {
if (is.null(booster$raw))
booster$raw <- xgb.save.raw(booster$handle)
}
if (class(booster) != "xgb.Booster.handle") {
stop("xgb.iter.update: first argument must be type xgb.Booster.handle")
}
if (class(dtrain) != "xgb.DMatrix") {
stop("xgb.iter.update: second argument must be type xgb.DMatrix")
}
.Call("XGBoosterBoostOneIter_R", booster$handle, dtrain, gpair$grad, gpair$hess,
.Call("XGBoosterBoostOneIter_R", booster, dtrain, gpair$grad, gpair$hess,
PACKAGE = "xgboost")
return(TRUE)
}
# iteratively update booster with dtrain
xgb.iter.update <- function(booster, dtrain, iter, obj = NULL) {
if (class(booster) != "xgb.Booster") {
stop("xgb.iter.update: first argument must be type xgb.Booster")
} else {
if (is.null(booster$handle)) {
booster$handle <- xgb.load(booster$raw)
} else {
if (is.null(booster$raw))
booster$raw <- xgb.save.raw(booster$handle)
}
if (class(booster) != "xgb.Booster.handle") {
stop("xgb.iter.update: first argument must be type xgb.Booster.handle")
}
if (class(dtrain) != "xgb.DMatrix") {
stop("xgb.iter.update: second argument must be type xgb.DMatrix")
}
if (is.null(obj)) {
.Call("XGBoosterUpdateOneIter_R", booster$handle, as.integer(iter), dtrain,
.Call("XGBoosterUpdateOneIter_R", booster, as.integer(iter), dtrain,
PACKAGE = "xgboost")
} else {
pred <- predict(booster, dtrain)

1
R-package/R/xgb.load.R
View File

@ -19,6 +19,7 @@
xgb.load <- function(modelfile) {
if (is.null(modelfile))
stop("xgb.load: modelfile cannot be NULL")
bst <- list(handle = NULL,raw = NULL)
class(bst) <- 'xgb.Booster'
bst$handle <- xgb.Booster(modelfile = modelfile)