执行: graph.write_png('0_DT_Tree/tree.png')时报错

测试决策树模型以下代码时，报错，

from sklearn.datasets import load_iris

from sklearn import tree

import os

import pydot # need install

print(os.getcwd())

clf = tree.DecisionTreeClassifier(criterion = "entropy") #gini

iris = load_iris()

 

clf = clf.fit(iris.data, iris.target)

tree.export_graphviz(clf, out_file='0_DT_Tree/tree.dot')        

(graph,) = pydot.graph_from_dot_file('0_DT_Tree/tree.dot')

graph.write_png('0_DT_Tree/tree.png')