R Ctree如何获得" n"在终端节点

时间:2015-11-08 19:28:12

标签: r

library("caret")
library("randomForest")
library("party")
library("tree")
library("rpart")
library("rpart.plot")
library("survAUC")
library("survival")
options(digits=4)    
survivalTree<-ctree(Surv(survData$user_age,survData$is_user_active)~.,data=survData)
plot(survivalTree)

在下面的代码中我创建了这个Ctree

数据集的示例

enter image description here

树看起来像这样:

enter image description here

我的问题是如何获得&#39; n&#39;终端节点中的变量..

让我们说节点4是终端,n = 17.我如何得到&#34; 17&#34;

的这个值?

1 个答案:

答案 0 :(得分:2)

where - 函数提取BinaryTree对象中的where插槽,您可以将其制表。我没有你的数据,但是使用类似的命令集来使用pkg中的卵巢数据:生存并得到:

 > where(survivalTree)
[1] 3 3 3 2 2 2 2 3 3 2 2 2 2 2 2 2 2 2 2 2 2 3 2 3 3 2

因此,要查看node-3中有多少个案例,它将是:

 stbl<- table( where(survivalTree) 
stbl[names(stbl)==3]
3 
8