当前位置:
首页
文章
前端
详情

Python算法之决策树利器——Graphviz

解密思想和方法,谁都会写程序!

Python算法之决策树利器——Graphviz

Python算法之决策树利器——Graphviz

(转载作者请注明出处)

废话不多说,直接上货....

Python算法之决策树利器——Graphviz

Python算法之决策树利器——Graphviz

Graphviz是开源的图形可视化软件。图形可视化是一种将结构信息表示为抽象图形和网络图的方式。它在网络,生物信息学,软件工程,数据库和网页设计,机器学习以及其他技术领域的可视界面中具有重要的应用。Python算法之决策树利器——Graphviz

一、Graphviz Windows环境安装

官网:https://graphviz.gitlab.io/\_pages/Download/Download\_windows.html

Python算法之决策树利器——Graphviz

进入windows命令行界面,输入dot -version,然后按回车,如果显示graphviz的相关版本信息,则安装配置成功。如图:

Python算法之决策树利器——Graphviz

配置环境变量同其他软件,我直接使用msi安装的,安装结果为:

Python算法之决策树利器——Graphviz

Python算法之决策树利器——Graphviz

我们可以新建一个gv文件:

Python算法之决策树利器——Graphviz

基本语法:

字符串 都要加双引号, 可以加\n换行符

注释 双斜杠// 或/* */

有向图 digraph, 节点关系: 指向->

无向图 graph, 节点关系: 连通 --

属性 node[attribute1=value1, attribute2=value2]

大小:size=”2,2”; 单位为英寸

标签:label=”显示在图上的内容”

边:edge [color=red,style=dotted]; 这句话之后生效

节点:node [color=navy]; 这句话之后生效

边方向:rankdir=参数值;LR(从左到右),RL,BT,TB

节点形状:a[shape=box]; 默认是椭圆

边框大小:a[width=.1,height=2.5]; 单位为英寸

边框颜色:a[color=red];

构造边

关系 有向图 无向图

一对一 a->b; a–b;

一对多 a->{b;c;d}; a–{b;c;d};

多对一 {b;c;d}->a; {b;c;d}–a;

多对多 {m,n,p,q}->{b;c;d}; {m,n,p,q}->{b;c;d};

详细资料:

官方文档:http://www.graphviz.org/documentation/

属性设置:https://graphviz.gitlab.io/\_pages/doc/info/attrs.html

节点形状:https://graphviz.gitlab.io/\_pages/doc/info/shapes.html

箭头形状:https://graphviz.gitlab.io/\_pages/doc/info/arrows.html

颜色配置:https://graphviz.gitlab.io/\_pages/doc/info/colors.html

二、在Python或PyCharm安装GV库

pip install graphviz 和conda install graphviz

或者自行找到PyCharm设置下编译环境安装库的位置搜索安装即可

Python算法之决策树利器——Graphviz

PyCharm中可调用

# coding:utf-8from graphviz import Digraphdot = Digraph(comment='The Round Table')# 添加圆点 A, A的标签是 King Arthurdot.node('A', 'king')dot.view()  #后面这句就注释了,也可以使用这个命令查看效果# 添加圆点 B, B的标签是 Sir Bedevere the Wisedot.node('B', 'Sir Bedevere the Wise')#dot.view()# 添加圆点 L, L的标签是 Sir Lancelot the Bravedot.node('L', 'Sir Lancelot the Brave')#dot.view()#创建一堆边,即连接AB的边,连接AL的边。dot.edges(['AB', 'AL'])#dot.view()# 在创建两圆点之间创建一条边dot.edge('B', 'L', constraint='false')#dot.view()# 获取DOT source源码的字符串形式print(dot.source)# 保存source到文件,并提供Graphviz引擎dot.render('test-output/round-table.gv', view=True)

执行结果:

// The Round Table

生成pdf形式:

Python算法之决策树利器——Graphviz

三、制作决策树

以下使用一个高尔夫的预测作为例子,使用的方法是sklearn 构建法

#决策树

**

执行结果:

Python算法之决策树利器——Graphviz

Python算法之决策树利器——Graphviz

参考文献

https://blog.csdn.net/mouday/article/details/80902992

Python算法之决策树利器——Graphviz

**

Python算法之决策树利器——Graphviz

本文分享自微信公众号 - AI科技与算法编程(kangsinx)。
如有侵权,请联系 support@oschina.cn 删除。
本文参与“OSC源创计划”,欢迎正在阅读的你也加入,一起分享。

免责申明:本站发布的内容(图片、视频和文字)以转载和分享为主,文章观点不代表本站立场,如涉及侵权请联系站长邮箱:xbc-online@qq.com进行反馈,一经查实,将立刻删除涉嫌侵权内容。