распечатка правил дерева, вычисленных из пакета randomForest от R - PullRequest
3 голосов
/ 01 мая 2011

Я использую пакет R randomForest.Есть ли способ для меня выяснить правила разделения, используемые на каждом узле в вычисляемых деревьях?

Пример кода:

library(randomForest)
mydata = data.frame(output = factor(c(0, 0, 0, 1, 1, 1)), x = c(0, 1, 0, 0, 1, 1), y = c(1, 1, 1, 0, 0, 1))
mydata.rf = randomForest(output ~ ., data = mydata, ntree = 3)

Я подозреваю, что это как-то связано с mydata.rf$forest$treemap, но я не уверен.

1 Ответ

3 голосов
/ 01 мая 2011

Ах, ответил сам: я могу использовать функцию getTree, описанную в pdf документации.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...