Как я могу извлечь решения из древовидного объекта rpart пакета R?

Я использую пакет rpart для моделирования дерева классификации/регрессии. У меня есть переменные x, y, s, где x находится в {-1,1}, y непрерывен в [0,1], а s $ является фактором с 3 уровнями.

я использую

fit <- rpart(x~y+s, data=data, method="class")

Окончательная модель имеет смысл, я могу построить ее, используя fancyRpartPlot(fit). Я также могу обрезать дерево, и остается только одно решение:

fit = prune(fit,cp=0.1) 

В нем говорится, что $y$ должно быть меньше некоторого порога. Как я могу извлечь этот порог? Я хотел бы выполнить начальную загрузку набора данных и сохранить порог в массиве.


person Richard    schedule 21.04.2015    source источник


Ответы (1)


Поле split в объекте rpart содержит границы решения. Решение

threshold =  fit$split[1,foo]
person Richard    schedule 21.04.2015