Python决策树图

2024-05-15 14:15:40 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图用scikit learn实现一个决策树,然后用Graphviz可视化这个树,我知道这是可视化DT的标准选择。我用的是PyCharm,anaconda,Python 2.7和OS X El Capitan。据我所知,我已经安装了pydot和Graphviz以及PIP install,并且还直接在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,但我无法查看该文件。 一。如何让graphviz存储库正常工作? 2。如何将图形写入PDF/PNG?我看到了一些例子,但没起作用 三。我找到这个命令:dot-Tps filename.dot-o outfile.ps 我在哪里用?如何验证我的OS X上是否存在dot实用程序?

提前谢谢!


Tags: fromimporttreeirisos可视化asload