17 lines
485 B
R
17 lines
485 B
R
#' @export
|
|
setClass("xgb.Booster")
|
|
|
|
#' @export
|
|
setMethod("predict",
|
|
signature = "xgb.Booster",
|
|
definition = function(object, newdata, outputmargin = FALSE)
|
|
{
|
|
if (class(newdata) != "xgb.DMatrix") {
|
|
newdata = xgb.DMatrix(newdata)
|
|
}
|
|
ret <- .Call("XGBoosterPredict_R", object, newdata,
|
|
as.integer(outputmargin), PACKAGE="xgboost")
|
|
return(ret)
|
|
})
|
|
|