决策树是一种非常有用的分类方法,它能够对新出现的对象给出正确的分类。比起文本描述的规则,我们更希望能使用图形来直观展示决策树的结果,这就是本文介绍的重点——决策树结果可视化。
使用rpart.plot包绘图函数rpart.plot()
优点:图形美观,规则显示清晰明了
缺点:只适用于rpart模型
rpart.plot()函数的内部参数结构如下:
以hospital数据为例,使用rpart()进行决策树模型的构建,并将结果保存到model变量中,在这个模型中,左端是因变量,右端的"."表示将除了use以外的所有变量作为自变量,你可以根据数据来调整自己的模型结构。之后用rpart.plot()进行可视化。
library(rpart.plot)
model <- rpart