决策树的可视化呈现

14 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

机器学习算法之Catboost

CatBoost是俄罗斯的搜索巨头Yandex在2017年开源的机器学习库,是Gradient Boostin
2 min read

机器学习算法之LightGBM

上一篇文章介绍了一个梯度提升决策树模型XGBoost,这篇文章我们继续学习一下GBDT模型的另一个进化版本:L
7 min read

机器学习算法之XGBoost

在上一篇Boosting方法的介绍中,对XGBoost有过简单的介绍。为了更还的掌握XGBoost这个工具。我
14 min read

发表评论

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