【问题标题】:R Ctree how to get the "n" in a terminal nodeR Ctree如何在终端节点中获取“n”
【发布时间】:2016-02-09 10:41:59
【问题描述】:
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

数据集示例

树看起来像这样:

我的问题是如何在终端节点中获取“n”变量..

假设节点 4 是终端,n = 17。我如何获得“17”这个值?

【问题讨论】:

    标签: r


    【解决方案1】:

    where-函数提取 BinaryTree 对象中的 where 槽,您可以将其制成表格。我没有你的数据,但是使用了一组类似的命令,在 pkg:survival 中使用卵巢数据集并获取:

     > 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
    

    所以要查看节点 3 中有多少案例,应该是:

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

    【讨论】:

    • 谢谢伙计。你是神! ;>
    猜你喜欢
    • 2012-12-07
    • 2015-07-12
    • 2014-02-21
    • 2015-08-19
    • 2015-09-09
    • 2012-07-12
    • 2017-11-25
    • 2015-04-22
    • 2016-08-12
    相关资源
    最近更新 更多