fix error

This commit is contained in:
El Potaeto 2015-01-11 23:39:24 +01:00
parent d441a9d382
commit 48c1911bc4

View File

@ -100,7 +100,7 @@ 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) break
if(length(tree) <2) next
treeID <- i-1