## 使用graph.write_png()出错 ##
最近在学习机器学习里面的决策树,看完理论了之后在网上找了鸢尾花的决策树实例来实现,但是到了最后一步graph.write_png(r"D:\python的学习笔记\somefile.png")
的时候出现 "dot.exe" not found in path
,查了之后发现是之前安装pydot和Graphviz时没有将Graphviz2.38添加到环境变量中,添加环境变量的代码如下:
import pydot
import os
os.environ['PATH'] = os.environ['PATH'] + (';D:\\Python\\Scripts\\graphviz-2.38\\release\\bin\\')
但是弄好环境变量之后用的代码是graph.write_png("somefile.png")
所以又出现[Errno 13] Permission denied: 'somefile.png'
这样的报错,所以我把代码改成了graph.write_png(r"D:\python的学习笔记\somefile.png")
就成功啦~~~~
希望能帮助到和我遇见同样问题的同学~~~
我再把我再网上找的鸢尾花的决策树demo附上吧…
from sklearn.datasets import load_iris
from sklearn import tree
from sklearn.tree import DecisionTreeClassifier, export_graphviz
import subprocess
import pydot
import os
os.environ['PATH'] = os.environ['PATH'] + (';D:\\Python\\Scripts\\graphviz-2.38\\release\\bin\\')
clf = tree.DecisionTreeClassifier()
iris = load_iris()
clf = clf.fit(iris.data, iris.target)
tree.export_graphviz(clf, out_file=r"D:\python的学习笔记\tree.dot")
(graph,) = pydot.graph_from_dot_file(r"D:\python的学习笔记\tree.dot")
graph.write_png(r"D:\python的学习笔记\somefile.png")
下面这个图片是最后的结果:
一起好好学习吧~~~