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
数据集的示例
树看起来像这样:
我的问题是如何获得&#39; n&#39;终端节点中的变量..
让我们说节点4是终端,n = 17.我如何得到&#34; 17&#34;
的这个值?答案 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