决策树的可视化呈现

12 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

使用GridSearchCV进行网格搜索

GridSearchCV简介 在机器学习模型中,需要人工选择的参数称为超参数。比如随机森林中决策树的个数,人工
49 sec read

PageRank算法学习与研究

什么是PageRank PageRank,简称PR,是Google排名运算法则(排名公式)的一部分,是Goog
2 min read

多经纬度坐标的中心点计算方法

在实际的应用场景,通常会遇到计算多个经纬度中心的需求。而在计算经纬度中心点通常有三种方式,每种方式对应不同的需
1 min read

发表评论

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