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

avoid error when a tree is just made of one leaf

Browse Source
This commit is contained in:
El Potaeto 2015-01-11 23:37:02 +01:00
parent 31a3b38ef8
commit d441a9d382
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
R-package/R/xgb.model.dt.tree.R
View File

@ -99,6 +99,9 @@ xgb.model.dt.tree <- function(feature_names = NULL, filename_dump = NULL, model
tree <- text[(position[i]+1):(position[i+1]-1)] tree <- text[(position[i]+1):(position[i+1]-1)]
# avoid tree made of a leaf only (no split)
if(length(tree) <2) break
treeID <- i-1 treeID <- i-1
notLeaf <- str_match(tree, "leaf") %>% is.na notLeaf <- str_match(tree, "leaf") %>% is.na
@ -129,7 +132,7 @@ xgb.model.dt.tree <- function(feature_names = NULL, filename_dump = NULL, model
} }
yes <- allTrees[!is.na(Yes),Yes] yes <- allTrees[!is.na(Yes),Yes]
set(allTrees, i = which(allTrees[,Feature]!= "Leaf"), set(allTrees, i = which(allTrees[,Feature]!= "Leaf"),
j = "Yes.Feature", j = "Yes.Feature",
value = allTrees[ID == yes,Feature]) value = allTrees[ID == yes,Feature])