我正在使用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
有关,但我不确定。
答案 0 :(得分:3)
啊,自己回答:我可以使用pdf文档中描述的getTree
函数。