找回密码
 立即注册

扫一扫,访问微社区

QQ登录

只需一步,快速开始

查看: 2666|回复: 0

[代码与实例] 利用pydot对决策树可视化时总出错,求大神指教

0

主题

0

帖子

0

积分

贫民

积分
0
吧啦吧啦 发表于 2018-11-8 15:46:53 | 显示全部楼层 |阅读模式
本帖最后由 吧啦吧啦 于 2018-11-8 15:50 编辑

from sklearn.datasets import load_iris
from sklearn import tree
iris = load_iris()
clf = tree.DecisionTreeClassifier()
clf = clf.fit(iris.data, iris.target)
from IPython.display import Image  
import pydot##########pydot的依赖包已经安装好
dot_data = StringIO()  
tree.export_graphviz(clf, out_file=dot_data,  
                         feature_names=iris.feature_names,  
                         class_names=iris.target_names,  
                         filled=True, rounded=True,  
                         special_characters=True)  
graph = pydot.graph_from_dot_data(dot_data.getvalue())  
Image(graph.create_png())


########################错误
AttributeError                            Traceback (most recent call last)<ipython-input-76-2c9a779a17ed> in <module>()      8                          special_characters=True)        9 graph = pydot.graph_from_dot_data(dot_data.getvalue())---> 10 Image(graph.create_png())AttributeError: 'list' object has no attribute 'create_png'
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

快速回复 返回顶部 返回列表