如何使用export_graphviz

使用Microsoft版本10，蟒蛇3.5.2，點顯示決策樹情節 - graphviz的版本2.38.0（正確安裝）

試圖用export_graphviz到可視化決策樹。認爲它非常接近，只是不能做最後一步。

這裏是示例代碼

from sklearn.datasets import load_iris 
from sklearn import tree 
clf = tree.DecisionTreeClassifier() 
iris = load_iris() 
clf = clf.fit(iris.data, iris.target) 
tree.export_graphviz(clf, out_file='tree.dot')

的 'tree.dot' 文件被輸出。當雙擊時，它會調用Microsoft Word並顯示以下文本。

digraph Tree { 
node [shape=box] ; 
0 [label="X[2] <= 2.45\ngini = 0.6667\nsamples = 150\nvalue = [50, 50, 50]"] ; 
1 [label="gini = 0.0\nsamples = 50\nvalue = [50, 0, 0]"] ; 
0 -> 1 [labeldistance=2.5, labelangle=45, headlabel="True"] ; 
2 [label="X[3] <= 1.75\ngini = 0.5\nsamples = 100\nvalue = [0, 50, 50]"] ; 
0 -> 2 [labeldistance=2.5, labelangle=-45, headlabel="False"] ; 
3 [label="X[2] <= 4.95\ngini = 0.168\nsamples = 54\nvalue = [0, 49, 5]"] ; 
2 -> 3 ; 
4 [label="X[3] <= 1.65\ngini = 0.0408\nsamples = 48\nvalue = [0, 47, 1]"] ; 
3 -> 4 ; 
5 [label="gini = 0.0\nsamples = 47\nvalue = [0, 47, 0]"] ; 
4 -> 5 ; 
6 [label="gini = 0.0\nsamples = 1\nvalue = [0, 0, 1]"] ; 
4 -> 6 ; 
7 [label="X[3] <= 1.55\ngini = 0.4444\nsamples = 6\nvalue = [0, 2, 4]"] ; 
3 -> 7 ; 
8 [label="gini = 0.0\nsamples = 3\nvalue = [0, 0, 3]"] ; 
7 -> 8 ; 
9 [label="X[0] <= 6.95\ngini = 0.4444\nsamples = 3\nvalue = [0, 2, 1]"] ; 
7 -> 9 ; 
10 [label="gini = 0.0\nsamples = 2\nvalue = [0, 2, 0]"] ; 
9 -> 10 ; 
11 [label="gini = 0.0\nsamples = 1\nvalue = [0, 0, 1]"] ; 
9 -> 11 ; 
12 [label="X[2] <= 4.85\ngini = 0.0425\nsamples = 46\nvalue = [0, 1, 45]"] ; 
2 -> 12 ; 
13 [label="X[1] <= 3.1\ngini = 0.4444\nsamples = 3\nvalue = [0, 1, 2]"] ; 
12 -> 13 ; 
14 [label="gini = 0.0\nsamples = 2\nvalue = [0, 0, 2]"] ; 
13 -> 14 ; 
15 [label="gini = 0.0\nsamples = 1\nvalue = [0, 1, 0]"] ; 
13 -> 15 ; 
16 [label="gini = 0.0\nsamples = 43\nvalue = [0, 0, 43]"] ; 
12 -> 16 ; 
}

此示例代碼工作正常

[http://scikit-learn.org/stable/auto_examples/tree/plot_iris.html#sphx-glr-auto-examples-tree-plot-iris-py] [1]

在此先感謝

來源

2016-12-12 bud fox

這個答案對我來說真是棒極了！謝謝阿什利！

對於windows：dl msi並安裝;在程序列表中找到gvedit.exe;打開有問題的.dot文件;點擊工具欄上的運行人員;轉到圖形 - >設置;將輸出文件類型更改爲您喜歡的文件類型，然後按OK。它不會說任何內容，只需在與.dot文件相同的目錄中找到該文件即可。 - 阿什利15年3月26日在9:15

Graphviz: How to go from .dot to a graph?

來源

2016-12-27 14:13:16

如何使用export_graphviz

回答

相關問題