我正在使用R来制作回归树模型:
library(party)
mydata = read.csv(“data.csv”)
TreeModel = ctree(price ~., data = mydata)
我想从根到任何叶子提取树的节点。但我找不到任何功能来做到这一点。 例如,如果树如下所示: 我想得到从根到任何叶子的路径。所以对于右边的第一片叶子来说我想得到一条路径(Koc,AirDecat,OTW,OTW,AirDecay),对于最左边的叶子,它应该是(Koc,AirDecay)。任何提示都非常感激。
答案 0 :(得分:1)
正如MrFlick所说,你应该提供一个可重复的例子。
这可能会让您开始了解如何找到聚会包的BinaryTree对象的路径:
library(data.tree)
library(party)
airq <- subset(airquality, !is.na(Ozone))
airct <- ctree(Ozone ~ ., data = airq,
controls = ctree_control(maxsurrogate = 3))
CreateNodeFromParty <- function(splitNode) {
node <- Node$new(splitNode$nodeID,
weights = splitNode$weights,
criterion = splitNode$criterion,
psplit = splitNode$psplit)
if (!splitNode$terminal) {
node$AddChildNode( CreateNodeFromParty(splitNode$left) )
node$AddChildNode( CreateNodeFromParty(splitNode$right) )
}
return (node)
}
tree <- CreateNodeFromParty(airct@tree)
tree
这将为您提供data.tree结构:
levelName
1 1
2 ¦--2
3 ¦ ¦--3
4 ¦ °--4
5 ¦ ¦--5
6 ¦ °--6
7 °--7
8 ¦--8
9 °--9
要查找特定节点,请执行以下操作:
tree$FindNode(6)$path
哪个会给你:
[1] "1" "2" "4" "6"