2015-11-08 2 views
0
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) 

в следующем коде я создал этот CtreeR Ctree как получить «N» в терминальном узле

пример набора данных

enter image description here

Дерево выглядит это:

enter image description here

Мой вопрос заключается в том, как сделать Я получаю переменную «n» в терминальных узлах.

позволяет говорить в узле 4, который является терминальным, n = 17. Как мне получить это значение «17»?

ответ

1

where -function извлекает слот where объекта BinaryTree, и вы можете его табулировать. У меня нет данных, но с аналогичным набором команд используются данные яичника, установленных в упаке: выживании и получить:

> 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 
+0

Thanks mate. Ты бог! ;> –

Смежные вопросы