当前位置: 首页 > article >正文

如何画出漂亮的决策树?

今天的内容介绍如何将决策树模型画出来。

进入实战部分!

首先安装所需的R包,并且载入: 

install.packages("rpart")
install.packages("rpart.plot")

library(rpart)
library(rpart.plot)

R包get!

下一步,使用mtcars数据集建立一个决策树模型,其中的mpg作为因变量,代码如下: 

tree <- rpart(mpg ~ hp + wt + am, data = mtcars)

将上述模型tree进行作图: 

rpart.plot(tree)

图片

以右下角的方框内容为例,一个方框指代一个节点,其中的数字29指的是此决策树给出的预测值mpg=29,样本占总数的22%。

方框的颜色(这里为蓝色,可修改)与mpg的值成正比,即值越大,颜色越深。

可以通过代码改变决策树的展示风格,一共有6种风格可供选择,代码如下: 

par(mfrow = c(3, 2))
rpart.plot(tree, type = 0)
rpart.plot(tree, type = 1)
rpart.plot(tree, type = 2) #默认
rpart.plot(tree, type = 3)
rpart.plot(tree, type = 4)
rpart.plot(tree, type = 5)
par(mfrow = c(1, 1))

图片

大家可以根据个人的喜好进行选择。

还可以添加一些额外的信息,比如每个节点的样本量,代码如下: 

rpart.plot(tree, extra = 101) # 添加样本量

图片

也可以调整小数点保留的位数: 

rpart.plot(tree, digits = 5)

图片

最后,调节上述节点的颜色为橘红色,代码如下: 

rpart.plot(tree, box.palette = "Oranges")

图片

关于其它更多修饰,可以询问R的帮助系统:?rpart.plot

好啦,今天的内容就到这里。如果有帮助,记得分享给需要的人

图片

参考文献

https://cran.r-project.org/web/packages/rpart.plot/

http://www.kler.cn/a/415610.html

相关文章:

  • 音视频流媒体直播/点播系统EasyDSS互联网视频云平台介绍
  • Ollama - 简化使用本地大语言模型
  • [CISCN2019 华东南赛区]Web4
  • TS 使用泛型和 typeof约束参数
  • deepin 安装 chrome 浏览器
  • 02-Linux系统权限维持
  • 【maven-4】IDEA 配置本地 Maven 及如何使用 Maven 创建 Java 工程
  • 自动类型推导(auto 和 decltype);右值引用和移动语义
  • mysql8.0基础-锁基础(七)
  • neo4j desktop版命令行中导入导出dump
  • Unity之一键创建自定义Package包
  • 题目 3209: 蓝桥杯2024年第十五届省赛真题-好数
  • 信息学奥赛一本通 1448:【例题1】电路维修 | 洛谷 P4667 [BalticOI 2011 Day1] Switch the Lamp On 电路维修
  • 《使用Python进行数据挖掘:理论、应用与案例研究》
  • spine 动画层 动态权重
  • brew安装mongodb和php-mongodb扩展新手教程
  • 智启未来 扬帆5G:江苏移动打造“5G + 智慧教育”典范,引领教育新风尚
  • 个人博客接入github issue风格的评论,utteranc,gitment
  • Nuxt.js 应用中的 render:response 事件钩子
  • 【Java面试题】消息队列中,如何保证消息的顺序性?
  • SQL进阶——子查询与视图
  • Prophet时间序列算法总结及python实现案例
  • 关于Spring @Transactional事务传播机制详解
  • 前端面试题-1(详解事件循环)
  • Cesium CZML绘制Entity
  • 深入理解 MongoDB:一款灵活高效的 NoSQL 数据库