hxd1011 hxd1011 - 9 months ago 146
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