r - 在 randomForest 包中绘制 500 棵树中的一棵

Question

如何在 R 中同名包中的 randomForest 函数的输出中绘制树？例如，我使用 iris 数据并想要绘制 500 个输出树中的第一棵树。我的代码是

model <-randomForest(Species~.,data=iris,ntree=500)

Answer 1

您可以使用randomForest包中的getTree()函数(官方指南:https://cran.r-project.org/web/packages/randomForest/randomForest.pdf)

在iris数据集上:

require(randomForest)
data(iris)

## we have a look at the k-th tree in the forest
k <- 10
getTree(randomForest(iris[, -5], iris[, 5], ntree = 10), k, labelVar = TRUE)