Rodrigo Ferreira Rodrigues commited on
Commit
937d2a1
·
1 Parent(s): d16fc41

Correcting problem in parsing

Browse files
Files changed (1) hide show
  1. regression_evaluate.py +3 -5
regression_evaluate.py CHANGED
@@ -105,12 +105,10 @@ class regression_evaluate(evaluate.Metric):
105
 
106
  for gen, gold in zip(generations, golds):
107
  f_gold = list(set([float(g) for g in gold]))
108
- try:
109
- f_ans = ast.literal_eval(gen)
110
- except:
111
- f_ans = re.findall(r'\d+(?:\.\d+)?', gen)
112
 
113
- f_ans = list(set([float(a) for a in f_ans])) #get rid of duples values
114
 
115
  precision, recall = self._calculate_pre_rec(f_ans, f_gold, lambda x,y: abs(x-y))
116
 
 
105
 
106
  for gen, gold in zip(generations, golds):
107
  f_gold = list(set([float(g) for g in gold]))
108
+
109
+ f_ans = re.findall(r'\d+(?:,\d+)*(?:\.\d+)?|\d+(?:\.\d+)?', gen)
 
 
110
 
111
+ f_ans = list(set([float(a.replace(',', '')) for a in f_ans])) #get rid of duples values
112
 
113
  precision, recall = self._calculate_pre_rec(f_ans, f_gold, lambda x,y: abs(x-y))
114