From f617f3e41c500325f7d093111d8b6e65ec379aa4 Mon Sep 17 00:00:00 2001 From: Aleksey Filippov Date: Wed, 12 Mar 2025 13:30:58 +0400 Subject: [PATCH] Fix classification value type --- dt-cart/backend/tree/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/dt-cart/backend/tree/__init__.py b/dt-cart/backend/tree/__init__.py index 1aee135..e36523e 100644 --- a/dt-cart/backend/tree/__init__.py +++ b/dt-cart/backend/tree/__init__.py @@ -35,7 +35,7 @@ def get_rules( rules.append(Rule(antecedent, value)) # type: ignore else: index = np.argmax(tree_.value[node][0]) - value = float(classes[index]) + value = str(classes[index]) rules.append(Rule(antecedent, value)) # type: ignore recurse(0, antecedent, rules) @@ -87,7 +87,7 @@ def get_tree( nodes.append(TreeNode(parent, None, node, "result", "=", value)) else: index = np.argmax(tree_.value[node][0]) - value = float(classes[index]) + value = str(classes[index]) nodes.append(TreeNode(parent, None, node, "result", "=", value)) recurse(0, None, nodes)