Skip to content
Snippets Groups Projects
Commit 2c7c2ff0 authored by tomrink's avatar tomrink
Browse files

snapshot...

parent 2c4f9cf8
Branches
No related tags found
No related merge requests found
......@@ -190,7 +190,7 @@ def decision_tree(x_train, y_train, x_test, y_test, criterion='entropy', max_dep
return DT
# Use this to plot the tree -----------------------------------------------------------
# export_graphviz(DT, out_file='tree.dot', filled=True, feature_names=params)
# export_graphviz(DT, out_file='tree.dot', filled=True, class_names=['no icing', 'icing'], feature_names=params)
# !dot -Tpng tree.dot -o tree.png
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment