我正在使用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有关,但我不确定。

最佳答案

嗯,我自己回答了:我可以使用pdf文档中介绍的getTree函数。

09-11 19:16
查看更多