2014-11-16 3 views
0

Мне нужно извлечь путь и вероятность каждого листа в дереве решений. Вот краткий пример работы с:Извлечь вероятности из деревьев решений

data(iris) 

model<-rpart(Species~., data=iris) 

summary(model) 

enter image description here

Я хотел бы быть в состоянии извлечь все эту информацию в целом здесь, такие как «предсказанного класс», «графы класса» и «вероятность «в формате, который я могу помещать в таблицу, а не просто читать ее на консоли.

Большое спасибо!

+1

'unclass (model)' имеет много вещей, которые могут быть полезны. Похоже, если вы выберете то, что вы хотите, это может быть не так уж плохо. –

ответ

0

Вам необходимо получить доступ к свойствам «модели», являющимся экземпляром rpart.object. Документы: https://stat.ethz.ch/R-manual/R-devel/library/rpart/html/rpart.object.html

Например слот «где» имеет следующую информацию: целочисленный вектор такой же длины, что и число наблюдений в корневом узле, содержащий номер строки кадра, соответствующего листа узел, в который попадает каждое наблюдение.

Вероятности вы найдете в слот "frame".

+0

Это может быть правильный путь. Мне придется смотреть на него некоторое время, хотя это не совсем интуитивно. Вероятности, которые они перечисляют в модели $ frame $ yval2, не совпадают, что дано в резюме (модели) – Jason

+0

Я вижу те же вероятности в следующих столбцах модели $ frame: yval2.V5, yval2.V6, yval2.V7 –

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