From dc776827d95d2698a4897a0739e250190d0e8e0d Mon Sep 17 00:00:00 2001
From: Caroline DE POURTALES <cdepourt@montana.irit.fr>
Date: Tue, 5 Apr 2022 15:00:28 +0200
Subject: [PATCH] change fi with fname
---
pages/application/DecisionTree/utils/dtree.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/pages/application/DecisionTree/utils/dtree.py b/pages/application/DecisionTree/utils/dtree.py
index 1eb9de3..79088c2 100644
--- a/pages/application/DecisionTree/utils/dtree.py
+++ b/pages/application/DecisionTree/utils/dtree.py
@@ -174,7 +174,7 @@ class DecisionTree():
for line in lines[1:]:
feat, val, real = line.split()
- self.fvmap[tuple([feat, int(val)])] = '{0}{1}'.format(feat, real)
+ self.fvmap[tuple([feat, int(val)])] = '{0}{1}'.format(self.feature_names[feat], real)
#if feat not in self.feids:
# self.feids[feat] = len(self.feids)
--
GitLab