Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

only replace tabulation which begins a line (avoid wrong replacement in feature name)

Browse Source
This commit is contained in:
El Potaeto 2015-01-09 11:06:56 +01:00
parent 3e1eea0eea
commit 10f755e055
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
R-package/R/xgb.dump.R
View File

@ -40,8 +40,10 @@ xgb.dump <- function(model, fname = NULL, fmap = "", with.stats=FALSE) {
}
result <- .Call("XGBoosterDumpModel_R", model, fmap, as.integer(with.stats), PACKAGE = "xgboost")
if(is.null(fname)) return(str_split(result, "\n") %>% unlist %>% str_replace_all("\t"," ") %>% Filter(function(x) x != "", .))
writeLines(result, fname)
TRUE
}
if(is.null(fname)) {
return(str_split(result, "\n") %>% unlist %>% str_replace("^\t+","") %>% Filter(function(x) x != "", .))
} else {
writeLines(result, fname)
return(TRUE)
}
}