打印出从R的randomForest包计算出的树规则

时间:2011-05-01 00:14:56

标签: r

我正在使用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 个答案:

答案 0 :(得分:3)

啊,自己回答:我可以使用pdf文档中描述的getTree函数。