hxd1011 hxd1011 - 1 year ago 292
R Question

How can I get the depth of the tree in RPART model?

Suppose I build a toy tree model with RPART, how can I get the depth of the tree?

fit=rpart(factor(am)~.,mtcars,control=rpart.control(cp=0,minsplit = 1))

I know how to count the leaves, for binary tree, we can approximate the depth by number of leaves, but it is not directly the depth of the tree.


Answer Source

rpart has a unexported function tree.depth that gives the depth of each node in the vector of node numbers passed to it. Using data from the question:

nodes <- as.numeric(rownames(fit$frame))
## [1] 2
Recommended from our users: Dynamic Network Monitoring from WhatsUp Gold from IPSwitch. Free Download