我是靠谱客的博主 光亮夏天,最近开发中收集的这篇文章主要介绍将sklearn生成的决策树进行图形化展示 AttributeError: 'list' object has no attribute 'write_*',觉得挺不错的,现在分享给大家,希望可以做个参考。

概述

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_*'

解决方法:
http://www.cnblogs.com/iamxyq/p/5918611.html


最后

以上就是光亮夏天为你收集整理的将sklearn生成的决策树进行图形化展示 AttributeError: 'list' object has no attribute 'write_*'的全部内容,希望文章能够帮你解决将sklearn生成的决策树进行图形化展示 AttributeError: 'list' object has no attribute 'write_*'所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(61)

评论列表共有 0 条评论

立即
投稿
返回
顶部