Merge pull request #142 from pommedeterresautee/master

avoid warning message when a tree is just made of one leaf
This commit is contained in:
Tianqi Chen 2015-01-11 16:02:56 -08:00
commit 166e7525da

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)]
# avoid tree made of a leaf only (no split)
if(length(tree) <2) next
treeID <- i-1
notLeaf <- str_match(tree, "leaf") %>% is.na