r - 从 party 包中的回归树中获取节点的统计信息

Question

我在 R 中使用 party 包。

我想从结果树的各个节点获取各种统计数据(均值、中位数等)，但我不知道如何执行此操作。例如

airq <- subset(airquality, !is.na(Ozone))
airct <- ctree(Ozone ~ ., data = airq, 
                   controls = ctree_control(maxsurrogate = 3))
airct
plot(airct)

结果是具有 4 个终端节点的树。我如何获得每个节点的平均空气质量？

Answer 1

这比我想象的要难得多。尝试这样的事情:

a <- by(airq,where(airct),colMeans) #or whatever function you desire for colMeans
a
a$"3" #access at node three
a[["3"]] #same thing

您可能会通过 ?`BinaryTree-class` 找到其他一些有用的示例。