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

fix small bug introduced in refactoring

Browse Source
This commit is contained in:
El Potaeto 2014-12-28 11:30:55 +01:00
parent 2154a160a3
commit 84fb89af70
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
R-package/R/xgb.importance.R
View File

@ -38,10 +38,9 @@ xgb.importance <- function(feature_names, filename_dump){
treeDump <- function(feature_names, text){ treeDump <- function(feature_names, text){
result <- c() result <- c()
for(line in text){ for(line in text){
m <- regexec("\\[f.*\\]", line) p <- regexec("\\[f.*\\]", line) %>% regmatches(line, .)
p <- regmatches(line, m)
if (length(p[[1]]) > 0) { if (length(p[[1]]) > 0) {
splits <- sub("\\]", "", sub("\\[f", "", p[[1]])) %>% strsplit("<")[[1]] %>% as.numeric splits <- sub("\\[f", "", p[[1]]) %>% sub("\\]", "", .) %>% strsplit("<") %>% .[[1]] %>% as.numeric
result <- c(result, feature_names[splits[1]+ 1]) result <- c(result, feature_names[splits[1]+ 1])
} }
} }
@ -50,5 +49,5 @@ treeDump <- function(feature_names, text){
} }
linearDump <- function(feature_names, text){ linearDump <- function(feature_names, text){
which(text == "weight:") %>% {a=.+1; text[a:length(text)]} %>% as.numeric %>% data.table(Feature = feature_names, Weight = .) which(text == "weight:") %>% {a=.+1;text[a:length(text)]} %>% as.numeric %>% data.table(Feature = feature_names, Weight = .)
} }