python决策树GraphViz可视化

安装GraphViz

下载地址:https://graphviz.gitlab.io/_pages/Download/Download_windows.html 选择msi文件下载,安装即可。安装完成后,将bin目录加入到系统path环境变量中

安装graphviz的python库

使用conda install python-graphviz命令安装即可

生成可视化文件

import graphviz # doctest: +SKIP
from sklearn import tree
print(data.columns)
#ID3为决策树分类器fit之后得到的模型,注意这里必须在fit后执行,在predict之后运行会报错
dot_data = tree.export_graphviz(ID3, out_file=None,feature_names=data.columns[:-1],class_names=np.unique(y)) # doctest: +SKIP
graph = graphviz.Source(dot_data) # doctest: +SKIP
#在同级目录下生成tree.pdf文件
graph.render("tree") # doctest: +SKIP

参考http://sklearn.apachecn.org/cn/0.19.0/modules/tree.html#tree-algorithms

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏nummy

Jupyter notebook 快速入门

在浏览器中输入http://localhost:8888,就出现了notebook的页面。

8120
来自专栏IMWeb前端团队

针对Vue的后台权限功能实现思路(持续更新)

权限是一块设计挺繁琐的功能,尤其是设计到前端SPA应用,前后端的耦合性太强,先屡屡思路,再实现,如果您有好的建议,也可评论留言。 基本的表结构如下 用户表。...

27350
来自专栏云计算认知升级

【腾讯云的1001种玩法】 十分钟轻松搞定云架构 · 负载均衡的几种均衡模式

视频内容 [fyckc.jpeg] 今天,我们来学习一下负载均衡的几种均衡模式。通过了解负载均衡的均衡模式,我们可以更好的利用负载均衡来为我们的应用服务。 ...

64060
来自专栏ChaMd5安全团队

挖洞经验之代理不当日进内网

大家好,我是STCX,应M姐姐之邀写一篇文章为咱们公众号做点贡献,那我就扯一下之前挖一个漏洞的经验。 ---- 正向代理和反向代理是forward/revers...

35760
来自专栏码洞

一种简单的Failover机制

在应用结构上有这样一个业务场景,机房里部署了多个物理数据库的Proxy无状态节点,业务端通过Proxy节点间接和存储DB交互。Proxy支持了分库分表的特性,管...

18320
来自专栏小樱的经验随笔

BugkuCTF 你必须让他停下

29240
来自专栏乐沙弥的世界

Nginx内置状态信息(http_stub_status)

Nginx提供了一个内置的状态信息监控页面,可用于监控Nginx的整体访问情况。这个内置功能由模块ngx_http_stub_status_module实现。如...

10820
来自专栏云端漫步

harbor源码分析之配置文件生成脚本解析(三)

在harbor第二篇中介绍了安装脚本install.sh,在一个脚本中有一step是执行prepare脚本.这个脚本主要是用来生成配置文件的.在本篇中,将对这一...

14410
来自专栏Y大宽

基于RNA-seq的共表达网络分析工具

12720
来自专栏浪淘沙

JFreeChart与struts框架冲突解决

8320

扫码关注云+社区

领取腾讯云代金券