python调用scikit-learn库画出决策树的图结构
大神们,调sklearn库画出自带数据集的决策树图结构的代码中出现以下的问题,查了之后没找到相关资料,能赐教下不,非常感谢
if __name__ == '__main__':
iris = load_iris()
exampleList=iris.data
labelList=iris.target
clf=tree.DecisionTreeClassifier()
mytree=clf.fit(exampleList, labelList)
dot_data = StringIO()
tree.export_graphviz(mytree, out_file=dot_data)
graph= pydot.graph_from_dot_data(dot_data.getvalue())
graph.write_png('iris_simple.png')
为什么最后一行“graph.write_png('iris_simple.png')”报错说AttributeError: 'list' object has no attribute 'write_png,graph返回是一个list,我改为graph[0].write_png('iris_simple.png'),又报Exception: "dot.exe" not found in path.