diff --git a/pages/application/DecisionTree/utils/dtree.py b/pages/application/DecisionTree/utils/dtree.py index 1eb9de3978b04b026de42c81dc3b600eb8b35345..79088c2cce1cbb0c4c6e65183195a4a110f357f6 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)