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/xgb.load.R
+++ b/R-package/R/xgb.load.R
@@ -21,7 +21,12 @@ xgb.load <- function(modelfile) {
    stop("xgb.load: modelfile cannot be NULL")
  
  handle <- xgb.Booster(modelfile = modelfile)
-  bst <- xgb.handleToBooster(handle)
+  # re-use modelfile if it is raw so we donot need to serialize
+  if (typeof(modelfile) == "raw") {
+    bst <- xgb.handleToBooster(handle, modelfile)
+  } else {
+    bst <- xgb.handleToBooster(handle, NULL)
+  }
  bst <- xgb.Booster.check(bst)
  return(bst)
 }