Graphviz 决策树

Question

我想在我的数据集上可视化应用的决策树分类器，看看它是如何分支的。做了几次谷歌搜索，这个链接出现了。我对这些代码很好，直到这一行“f = tree.export_graphviz(clf, out_file=f)”。

from sklearn.datasets import load_iris
from sklearn import tree
iris = load_iris()
clf = tree.DecisionTreeClassifier()
clf = clf.fit(iris.data, iris.target)
with open("iris.dot", 'w') as f:
...     f = tree.export_graphviz(clf, out_file=f)

我的问题是在这段代码之后我如何可视化树？根据链接“http://scikit-learn.org/stable/modules/tree.html”，我必须使用此代码“dot -Tpdf iris.dot -o iris.pdf”来创建 PDF 文件。我不明白我应该在哪里使用它？在 Graphviz 的点工具中？如果是，我会收到此错误“错误：：'点'附近的第 1 行中的语法错误”

如果有人回答我的问题，我将不胜感激。谢谢。

Answer 1

dot -Tpdf iris.dot -o iris.pdf 是一个可以在 bash 中使用的命令。并且你应该已经安装了 Graphviz 工具。例如，你可以在 ubuntu 上安装它使用命令：sudo apt-get install graphviz

根据链接“http://scikit-learn.org/stable/modules/tree.html”，如果我们安装了 Python 模块 pydotplus，我们可以直接在 Python 中生成 PDF 文件（或任何其他支持的文件类型）：

import pydotplus 
dot_data = tree.export_graphviz(clf, out_file=None) 
graph = pydotplus.graph_from_dot_data(dot_data) 
graph.write_pdf("iris.pdf") 

对于Shelly的评论，我添加以下代码，这是我的ipython笔记本上运行的完整代码。

import pydotplus 
from sklearn.datasets import load_iris
from sklearn import tree
iris = load_iris()
clf = tree.DecisionTreeClassifier()
clf = clf.fit(iris.data, iris.target)
with open("iris.dot", 'w') as f:
    f = tree.export_graphviz(clf, out_file=f)

dot_data = tree.export_graphviz(clf, out_file=None) 
graph = pydotplus.graph_from_dot_data(dot_data) 
graph.write_pdf("iris.pdf")