在R的ggplot中绘制树

Question

我想知道是否有可能通过ggplot绘制树？ 让我们说：

library(rpart
library(rpart.plot)

data(iris)
mod <- rpart(Species~., data=iris)
prp(mod)

我可以在ggplot中绘制类似的图吗？

Answer 1

autoplot.rpart()包中的survMisc功能可以帮助您实现目标。但是你可能需要清理情节的表现形式，可能会以符号等形式分层。这似乎只是一个起点：

library(rpart)
library(survMisc)

data(iris)
mod <- rpart(Species~., data=iris)

autoplot(mod, branch=0)

Answer 2

您可以使用 rattle ，它会生成漂亮的rpart图

library(rattle)

fancyRpartPlot(rpart(Species~., data=iris),yesno=2,split.col="black",nn.col="black", 
               caption="",palette="Set2",branch.col="black")