我正在尝试使用 scikit learn 实现决策树,然后使用 Graphviz 可视化该树,我认为这是可视化 DT 的标准选择。我正在使用 PyCharm、anaconda、Python 2.7 和 OS X El Capitan。据我所知,我已经通过 PIP 安装安装了 pydot 和 Graphviz,并且还直接在 Pycharm 中安装了它们,但无论我做什么,我都会不断收到“没有名为 graphviz 的模块”。
from sklearn.datasets import load_iris
from sklearn import tree
#import graphviz as gv
# uncommenting the row above produces an error
clf = tree.DecisionTreeClassifier()
iris = load_iris()
clf = clf.fit(iris.data, iris.target)
with open('graph.dot', 'w') as file:
tree.export_graphviz(clf, out_file = file)
file.close()
目前运行此代码会生成 graph.dot 但我无法查看该文件。
1. 如何让 graphviz 存储库正常工作?
2. 如何将图表写入PDF/PNG?我看到了一些例子但没有用
3.我找到了这个命令:dot -Tps filename.dot -o outfile.ps
我在哪里使用它?如何验证我的 OS X 上是否存在 .t 实用程序?
提前致谢!
我很确定我使用自制程序安装了 graphviz,但看起来您也可以从下载二进制文件http://www.graphviz.org/Download_macos.php http://www.graphviz.org/Download_macos.php。如果你无法让 pydot 工作,你需要运行dot
来自终端的命令,或者在使用子进程的脚本中:
import subprocess
subprocess.call(['dot', '-Tpdf', 'tree.dot', '-o' 'tree.pdf'])
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)