我正在使用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
函数。