add handle and raw structure to xgb.Booster

2015-02-09 15:51:24 -08:00
parent ea5860d574
commit 5b611c355e
7 changed files with 46 additions and 16 deletions
--- a/R-package/R/xgb.load.R
+++ b/R-package/R/xgb.load.R
@@ -19,5 +19,9 @@
 xgb.load <- function(modelfile) {
  if (is.null(modelfile)) 
    stop("xgb.load: modelfile cannot be NULL")
-  xgb.Booster(modelfile = modelfile)
+  bst <- list(handle = NULL,raw = NULL)
+  class(bst) <- 'xgb.Booster'
+  bst$handle <- xgb.Booster(modelfile = modelfile)
+  bst$raw <- xgb.save.raw(bst$handle)
+  return(bst)
 }