python输出决策树图形的例子
windows10:
1,先要pip安装pydotplus和graphviz:
pipinstallpydotplus pipinstallgraphviz
2,www.graphviz.org下载msi文件并安装。
3,系统环境变量path中增加两项:
C:\ProgramFiles(x86)\Graphviz2.38\bin C:\ProgramFiles(x86)\Graphviz2.38 #确认graphviz是安装在上面路径当中。
4,python中使用方法:
fromsklearn.externals.siximportStringIO
importpydotplus
#drt是DecisionTreeClassifier(),在之前要fit训练之后才能在这里输出图形。
dot_data=StringIO()
tree.export_graphviz(drt,out_file=dot_data)
graph=pydotplus.graph_from_dot_data(dot_data.getvalue())
graph.write_png("out.png")#当前文件夹生成out.png
#这三行代码可以生成pdf:
dot_data=tree.export_graphviz(drt,out_file=None)
graph=graphviz.Source(dot_data)
graph.render()
以上这篇python输出决策树图形的例子就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持毛票票。
热门推荐
1 训练狗狗基础
2 怎么训练狗狗笑
3 怎么训练狗狗卧下
4 上海狗狗训练基地
5 训练狗狗排便
6 狗狗基础训练
8 如何训练狗狗装死
9 狗狗站姿训练
10 狗狗训练用品
11 训练狗狗吧
12 狗狗要怎么训练
13 怎么训练狗狗玩球
14 训练狗狗唤回
15 狗狗训练躺下
16 狗狗训练指令
17 狗狗的训练技巧
18 狗狗不好训练