is.ulstu.ru/ds-services
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix classification value type

Browse Source
This commit is contained in:
Aleksey Filippov 2025-03-12 13:30:58 +04:00
parent 5817314562
commit f617f3e41c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
dt-cart/backend/tree/__init__.py
View File

@ -35,7 +35,7 @@ def get_rules(
rules.append(Rule(antecedent, value)) # type: ignore rules.append(Rule(antecedent, value)) # type: ignore
else: else:
index = np.argmax(tree_.value[node][0]) index = np.argmax(tree_.value[node][0])
value = float(classes[index]) value = str(classes[index])
rules.append(Rule(antecedent, value)) # type: ignore rules.append(Rule(antecedent, value)) # type: ignore
recurse(0, antecedent, rules) recurse(0, antecedent, rules)
@ -87,7 +87,7 @@ def get_tree(
nodes.append(TreeNode(parent, None, node, "result", "=", value)) nodes.append(TreeNode(parent, None, node, "result", "=", value))
else: else:
index = np.argmax(tree_.value[node][0]) index = np.argmax(tree_.value[node][0])
value = float(classes[index]) value = str(classes[index])
nodes.append(TreeNode(parent, None, node, "result", "=", value)) nodes.append(TreeNode(parent, None, node, "result", "=", value))
recurse(0, None, nodes) recurse(0, None, nodes)