snapshot...

modules/machine_learning/classification.py

@@ -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