如何计算rpart中特定节点中的观测值?

时间:2019-01-18 01:53:59

标签: r tree

我的目标是在每个节点的数据中找到每个观察值。我只能找出每个节点中的观测值数量,然后无法获得该节点中的哪些观测值(例如,数量(预计节点(0,0)属于))

N=500
episolon=rnorm( N, 0, 0.01 )
x1=rnorm(N, 0, sd=1)
x2=rnorm(N, 0, sd=1)
X=data.frame( x1, x2)
eta_x=1/2*x1+x2
Kappa_x=1/2*x1
w=rbinom(N, 1, 0.5)
treatment <- w
makeY=function(eta,Kappa ){
    Y=eta+1/2*(2*w-1)*Kappa+episolon
}
Y1=makeY(eta_x, Kappa_x )
Y=eta_x+1/2*(2*w-1)*Kappa_x+episolon
design1 <- data.frame( x1, x2, Y1, treatment )
tree <- causalTree(Y1~x1+x2, data=design1, treatment=design1$treatment,
                  split.Rule = "TOT", cv.option = "fit", cv.Honest = F,
                  split.Bucket = T, xval=10, propensity = 0.5, minsize = 25)
opcp <- tree$cptable[,1][which.min(tree$cptable[,4])]
optree <- prune(tree, cp=opcp)
rpart.plot(optree)
number <- rpart.predict.leaves(opTree, newdata = data.frame(x1=0,x2=0), type="where")  

我尝试了2年前发布的其他方法,但我发现结果不合理。
是否有其他方法可以通过其他方式获得正确的结果?

0 个答案:

没有答案