假设我用 RPART 建立一个玩具树模型,我怎样才能得到树的深度?
library(rpart)
library(partykit)
fit=rpart(factor(am)~.,mtcars,control=rpart.control(cp=0,minsplit = 1))
plot(as.party(fit))
我知道如何计算叶子,对于二叉树,我们可以通过叶子的数量来近似深度,但它不直接是树的深度。
sum(fit$frame$var=="<leaf>")
最佳答案
rpart 有一个未导出的函数 tree.depth
,它给出传递给它的节点编号向量中每个节点的深度。使用问题中的数据:
nodes <- as.numeric(rownames(fit$frame))
max(rpart:::tree.depth(nodes))
## [1] 2
关于r - 如何获得 RPART 模型中树的深度?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/40897081/