Spyder中pydot的安装
要安装pydot,得首先安装Graphviz和pyparsing。
1. 安装graphviz(安装过程中注意选择for everyone);
(1)下载graphviz-2.38.msi,下载地址:https://graphviz.gitlab.io/_pages/Download/Download_windows.html
(2)安装完设置环境变量中path设置,在path中设置graphviz安装包中bin所在的路径,如“D:\Program Files\graphviz-2.26.3/bin”,否则pydot解析不出来。
(3)验证graphviz是否安装完成,在cmd中输入dot -version命令;
2. 安装pyparsing,打开Anaconda Prompt,通过命令行pip install pyparsing或conda install pyparsing进行安装;通过pip list或者conda list进行版本的查看。
3. 安装pydot,在Anaconda Prompt中通过命令行pip install pydot或者conda install pydot进行安装pydot
4. 测试,使用spyder进行一下脚本的测试。
from sklearn.datasets import load_iris #skit-learn是机器学习的开源包
from sklearn import tree
from sklearn.externals.six import StringIO
import pydot
iris = load_iris()#载入数据集
clf = tree.DecisionTreeClassifier()#算法模型
clf = clf.fit(iris.data, iris.target)#模型训练
dot_data = StringIO()
tree.export_graphviz(clf, out_file=dot_data)
graph = pydot.graph_from_dot_data(dot_data.getvalue())
graph[0].write_pdf("iris.pdf")#写入pdf
运行成功后将生成iris.pdf文件