决策树的绘制

在使用 sklean.tree.DecisionTreeClassifier 训练决策树,并画出决策树

1.按顺序安装一下包及程序:

(1)cmd   ->  pip install GraphViz

(2)在http://www.graphviz.org/Download_windows.php下载GraphViz的zip包,解压放在某目录下 (如果你不小心关掉安装完成之后自动打开的编辑器,可以重新运行下面这个EXE:D:\Program Files\Graphviz2.38\bin\gvedit.exe

(3)将系统路径的path后添加   D:\Program Files\Graphviz2.38\bin            

(4)pip install  pydotplus(python 2.7以上安装pydotplus 而非pydot)

(5)重启pyhton IDE

2.绘制决策树的程序

大概程序如此,不同的python版本可能会有所不同

from sklearn import tree
from IPython.display import display, Image
# from IPython.display import Image
# from sklearn.externals.six import StringIO
import pydotplus
import sys
reload(sys)
sys.setdefaultencoding('utf8')

dot_data = tree.export_graphviz(dt_clf2,out_file=None,filled=True,rounded=True,special_characters=True)  
graph = pydotplus.graph_from_dot_data(dot_data)  
display(Image(graph.create_png()))

参考博客:https://www.cnblogs.com/pinard/p/6056319.html

 

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值