|
本帖最后由 吧啦吧啦 于 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'
|
|