决策树的可视化呈现

13 sec read

上一篇决策树学习的文章中讲到决策树其中的一个优点是决策树模型可以可视化,非常直观。今天就一起来学习下如何使用决策树将模型可视化出来。

决策树的可视化一般使用的工具是Graphviz。在使用Graphviz之间还是有些门道。具体流程为:

  1. 安装Graphviz的可执行包,并在环境变量的PATH添加安装路径
  2. 安装需要的Python库: pip install graphviz ,并在代码中使用graphviz

方法一:使用export_graphviz 将树导出为 Graphviz 格式

这里生成的iris.dot是一个纯文本文件,可直接打开查看,但是为了更好的可视化,可以使用graphviz可执行包中的dot程序将其转化为可视化的PDF文档。具体方法为执行如下命令: dot -Tpdf iris.dot -o iris.pdf

方法二:使用 pydotplus 直接生成 iris.pdf

先安装pydotplus的Python库: pip install pydotplus ,然后就可以执行执行如下代码生成PDF文档了:

方法三:直接在 jupyter notebook 中生成

打赏作者
微信支付标点符 wechat qrcode
支付宝标点符 alipay qrcode

Windows安装PostgreSQL/MongoDB问题

PostgreSQL 相较于MySQL(MariDB)我更喜欢PostgreSQL,最近在Windows安装的
43 sec read

TensorFlow学习笔记:特征工程

特征工程是机器学习流程中重要的一个环节,即使是通常用来做端到端学习的深度学习模型在训练之前也免不了要做一些特征
5 min read

CTR预估模型FM、FFM、DeepFM

点击率(click-through rate, CTR) 是点击特定链接的用户与查看页面,电子邮件或广告的总用
6 min read

2 Replies to “决策树的可视化呈现”

  1. 博主你好,中文乱码如何解决?我在电脑上实验, jupyter notebook上面不会出现乱码,但是图无法保存

发表评论

电子邮件地址不会被公开。 必填项已用*标注