Spaces:
Sleeping
Sleeping
Rodrigo Ferreira Rodrigues
commited on
Commit
·
937d2a1
1
Parent(s):
d16fc41
Correcting problem in parsing
Browse files- 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 |
-
|
| 109 |
-
|
| 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 |
|