From 2c7c2ff0b63218d85bcd87d2f9980f304c71deb1 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Thu, 2 May 2024 10:04:46 -0500
Subject: [PATCH] snapshot...
---
modules/machine_learning/classification.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/modules/machine_learning/classification.py b/modules/machine_learning/classification.py
index d5fedef2..b0d64155 100644
--- a/modules/machine_learning/classification.py
+++ b/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
--
GitLab