allow xgb.load re-use raw information if necessary

2015-03-26 16:54:29 -07:00
parent 98618646f6
commit 4eae8e8676
2 changed files with 8 additions and 3 deletions
--- a/R-package/R/utils.R
+++ b/R-package/R/utils.R
@@ -77,9 +77,9 @@ xgb.Booster <- function(params = list(), cachelist = list(), modelfile = NULL) {
 }

 # convert xgb.Booster.handle to xgb.Booster
-xgb.handleToBooster <- function(handle)
+xgb.handleToBooster <- function(handle, raw = NULL)
 {
-  bst <- list(handle = handle, raw = NULL)
+  bst <- list(handle = handle, raw = raw)
  class(bst) <- "xgb.Booster"
  return(bst)
 }