概述
1,工具和平台:
python2.7 windows
2,决策树的可视化展示据我所知有两种途径:一是将生成的结果导出为pmml文件,工具包为sklearn2pmml等,具体可见https://github.com/jpmml/jpmml-sklearn 和https://github.com/jpmml/sklearn2pmml 如果遇到安装问题可以留言;而是进行graphviz的图形化展示,首先需要安装graphviz-2.38.msi url:http://www.graphviz.org/pub/graphviz/stable/windows/graphviz-2.38.msi,
这是因为python所用的工具包pydot依赖于dot.exe,配置
path:C:Program Files (x86)Graphviz2.38bin 和 C:Program Files (x86)Graphviz2.38 具体按照你自己的安装目录即可
接着安装pydot : pip install pydot
基本原理为:利用sklearn.tree.export_graphviz 可以将随机森林中每一棵树的结构以doc的形式表示出来,结合graphviz可以将结构图转为png或者pdf等。
#coding=utf-8
from sklearn.datasets import load_iris
from sklearn import tree
iris = load_iris()
clf = tree.DecisionTreeClassifier()
clf = clf.fit(iris.data, iris.target)
from sklearn.externals.six import StringIO
import pydot
dot_data = StringIO()
tree.export_graphviz(clf, out_file=dot_data)
graph = pydot.graph_from_dot_data(dot_data.getvalue())
graph[0].write_dot('iris_simple.dot')
graph[0].write_png('iris_simple.png')
可能会出现错误如:
AttributeError: 'list' object has no attribute 'write_*'
最后
以上就是光亮夏天为你收集整理的将sklearn生成的决策树进行图形化展示 AttributeError: 'list' object has no attribute 'write_*'的全部内容,希望文章能够帮你解决将sklearn生成的决策树进行图形化展示 AttributeError: 'list' object has no attribute 'write_*'所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。
发表评论 取消回复