fix save.raw

2015-02-09 17:35:50 -08:00 · 2015-02-09 17:35:50 -08:00 · 47b5cf5148
commit 47b5cf5148
parent 4c25600d2a
1 changed files with 3 additions and 0 deletions
--- a/R-package/R/xgb.save.raw.R
+++ b/R-package/R/xgb.save.raw.R
@ -18,6 +18,9 @@
 #' @export
 #' 
 xgb.save.raw <- function(handle) {
+  if (class(handle) == "xgb.Booster"){
+    handle <- handle$handle
+  }
  if (class(handle) == "xgb.Booster.handle") {
    raw <- .Call("XGBoosterModelToRaw_R", handle, PACKAGE = "xgboost")
    return(raw)