Skip to content

Commit 0b6a3b4

Browse files
committed
Added decision tree example beautiful graphviz
1 parent b2c176e commit 0b6a3b4

10 files changed

+150
-34
lines changed

.DS_Store

0 Bytes
Binary file not shown.

Sklearn/.DS_Store

0 Bytes
Binary file not shown.

Sklearn/DecisionTrees/.DS_Store

0 Bytes
Binary file not shown.

Sklearn/DecisionTrees/DecisionTreeAnatomy.ipynb

Lines changed: 137 additions & 33 deletions
Large diffs are not rendered by default.

Sklearn/DecisionTrees/iris_depth2_decisionTree.dot

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
digraph Tree {
22
node [shape=box] ;
3-
0 [label="petal width (cm) <= 0.8\ngini = 0.665\nsamples = 112\nvalue = [37, 34, 41]\nclass = virginica"] ;
3+
0 [label="petal length (cm) <= 2.35\ngini = 0.665\nsamples = 112\nvalue = [37, 34, 41]\nclass = virginica"] ;
44
1 [label="gini = 0.0\nsamples = 37\nvalue = [37, 0, 0]\nclass = setosa"] ;
55
0 -> 1 [labeldistance=2.5, labelangle=45, headlabel="True"] ;
66
2 [label="petal length (cm) <= 4.95\ngini = 0.496\nsamples = 75\nvalue = [0, 34, 41]\nclass = virginica"] ;
8.68 KB
Loading
Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
digraph Tree {
2+
node [shape=box] ;
3+
0 [label="petal length (cm) <= 2.35\ngini = 0.665\nsamples = 112\nvalue = [37, 34, 41]\nclass = virginica"] ;
4+
1 [label="gini = 0.0\nsamples = 37\nvalue = [37, 0, 0]\nclass = setosa"] ;
5+
0 -> 1 [labeldistance=2.5, labelangle=45, headlabel="True"] ;
6+
2 [label="petal length (cm) <= 4.95\ngini = 0.496\nsamples = 75\nvalue = [0, 34, 41]\nclass = virginica"] ;
7+
0 -> 2 [labeldistance=2.5, labelangle=-75, headlabel="False", arrowsize=1.7, penwidth = 3] ;
8+
3 [label="gini = 0.153\nsamples = 36\nvalue = [0, 33, 3]\nclass = versicolor"] ;
9+
2 -> 3 [labeldistance=2.5, labelangle=45, headlabel="True"] ;
10+
4 [label="gini = 0.05\nsamples = 39\nvalue = [0, 1, 38]\nclass = virginica"] ;
11+
2 -> 4 [labeldistance=2.5, labelangle=-75, headlabel="False", arrowsize=1.7, penwidth = 3];
12+
}
45.7 KB
Loading
342 KB
Binary file not shown.
490 KB
Loading

0 commit comments

Comments
 (0)