我有一个包含6个分类变量的数据集,其级别从5到28.我已经从 ctree()(聚会包)获得了具有17个终端节点的输出。我跟随ctree() - How to get the list of splitting conditions for each terminal node?的@Galled的输入来达到我想要的输出。
但是,运行代码后出现以下错误:
Error in data.frame(ResulTable, Means, Counts) :
arguments imply differing number of rows: 17, 2
我尝试添加这些额外的行:
ResulTable <- rbind(ResulTable, cbind(Node = Node, Path = Path2))
ResulTable$Node <- rownames(ResulTable)
melt(ResulTable)
到目前为止没有成功。关于它出错的地方的任何指示?
答案 0 :(得分:6)
我建议使用partykit
的新ctree()
实现而不是旧的party
包,然后您可以使用函数.list.rules.party()
。这尚未正式导出,但可用于提取所需信息。
library("partykit")
airq <- subset(airquality, !is.na(Ozone))
ct <- ctree(Ozone ~ ., data = airq)
partykit:::.list.rules.party(ct)
## 3 5
## "Temp <= 82 & Wind <= 6.9" "Temp <= 82 & Wind > 6.9 & Temp <= 77"
## 6 8
## "Temp <= 82 & Wind > 6.9 & Temp > 77" "Temp > 82 & Wind <= 10.3"
## 9
## "Temp > 82 & Wind > 10.3"