Получить правило / шаблон дерева решений для каждой строки прогнозируемого набора данных для пакета rpart / ctree в R
Я построил модель дерева решений в R, используяrpart
а такжеctree
, Я также предсказал новый набор данных, используя построенную модель, и получил предсказанные вероятности и классы.
Однако я хотел бы извлечь правило / путь в одну строку для каждого наблюдения (в прогнозируемом наборе данных). Сохраняя эти данные в табличном формате, я могу объяснить прогноз с помощью причины в автоматическом режиме, не открывая R.
Что означает, что я хочу получить следующее.
ObsID Probability PredictedClass PathFollowed
1 0.68 Safe CarAge < 10 & Country = Germany & Type = Compact & Price < 12822.5
2 0.76 Safe CarAge < 10 & Country = Korea & Type = Compact & Price > 12822.5
3 0.88 Unsafe CarAge > 10 & Type = Van & Country = USA & Price > 15988
Код, который я ищу, это
library(rpart)
fit <- rpart(Reliability~.,data=car.test.frame)
это то, что нужно расширить в несколько строк, возможно,
predResults <- predict(fit, newdata = newcar, type= "GETPATTERNS")