назначение узла rpart - PullRequest
Новая
       28

назначение узла rpart

3 голосов
/ 21 марта 2012

Можно ли извлечь назначение узла для подобранного дерева rpart?Как насчет того, когда я применяю модель к новым данным?

Идея состоит в том, что я хотел бы использовать узлы как способ кластеризации моих данных.В других пакетах (например, SPSS) я могу сохранить прогнозируемый класс, вероятности и номер узла для дальнейшего анализа.

Учитывая, насколько мощным может быть R, я думаю, что есть простое решение для этого.

1 Ответ

12 голосов
/ 21 марта 2012

Попробуйте использовать пакет partykit: 

library(rpart)
z.auto <- rpart(Mileage ~ Weight, car.test.frame)
library(partykit)
z.auto2 <- as.party(z.auto)
predict(z.auto2, car.test.frame[1:3,], type = "node")

# Eagle Summit 4 Ford Escort   4  Ford Festiva 4 
#              7               7               7
...