gpt4 book ai didi

r - 如何获得 RPART 模型中树的深度?

转载 作者:行者123 更新时间:2023-12-05 01:19:11 24 4
gpt4 key购买 nike

假设我用 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/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com