Spaces:
Sleeping
Sleeping
Rodrigo Ferreira Rodrigues commited on
Commit ·
1460313
1
Parent(s): 1b88c4f
Adding ast library
Browse files- coord_eval.py +3 -2
- tests.py +2 -2
coord_eval.py
CHANGED
|
@@ -16,6 +16,7 @@
|
|
| 16 |
import evaluate
|
| 17 |
import datasets
|
| 18 |
import re
|
|
|
|
| 19 |
|
| 20 |
|
| 21 |
# TODO: Add BibTeX citation
|
|
@@ -45,7 +46,7 @@ Returns:
|
|
| 45 |
accuracy: 1 if coordinates predicted are d distant from gold ones, O otherwise.
|
| 46 |
Examples:
|
| 47 |
>>> my_new_module = evaluate.load("rfr2003/coord_eval")
|
| 48 |
-
>>> results = my_new_module.compute(generations=["(12.7, 67.8)", "(16.7, 89.6)"], golds=[[12.7, 67.8], [10.9, 80.6]],
|
| 49 |
>>> print(results)
|
| 50 |
{'coord_accuracy': 0.5}
|
| 51 |
"""
|
|
@@ -119,7 +120,7 @@ class Coord_eval(evaluate.Metric):
|
|
| 119 |
if matches:
|
| 120 |
match = matches[0]
|
| 121 |
f_ans = (float(match[0].replace(',', '.').replace(' ', '')), float(match[3].replace(',', '.').replace(' ', '')))
|
| 122 |
-
correct +=
|
| 123 |
|
| 124 |
total += 1
|
| 125 |
|
|
|
|
| 16 |
import evaluate
|
| 17 |
import datasets
|
| 18 |
import re
|
| 19 |
+
import ast
|
| 20 |
|
| 21 |
|
| 22 |
# TODO: Add BibTeX citation
|
|
|
|
| 46 |
accuracy: 1 if coordinates predicted are d distant from gold ones, O otherwise.
|
| 47 |
Examples:
|
| 48 |
>>> my_new_module = evaluate.load("rfr2003/coord_eval")
|
| 49 |
+
>>> results = my_new_module.compute(generations=["(12.7, 67.8)", "(16.7, 89.6)"], golds=[[12.7, 67.8], [10.9, 80.6]], d_range=20)
|
| 50 |
>>> print(results)
|
| 51 |
{'coord_accuracy': 0.5}
|
| 52 |
"""
|
|
|
|
| 120 |
if matches:
|
| 121 |
match = matches[0]
|
| 122 |
f_ans = (float(match[0].replace(',', '.').replace(' ', '')), float(match[3].replace(',', '.').replace(' ', '')))
|
| 123 |
+
correct += self._accuracy_coord(f_ans, f_gold, d_range)
|
| 124 |
|
| 125 |
total += 1
|
| 126 |
|
tests.py
CHANGED
|
@@ -2,13 +2,13 @@ test_cases = [
|
|
| 2 |
{
|
| 3 |
"generations": ["(12.7, 67.8)", "(16.7, 89.6)"],
|
| 4 |
"golds": [[12.7, 67.8], [10.9, 80.6]],
|
| 5 |
-
"
|
| 6 |
"result": {"coord_accuracy": 0.5}
|
| 7 |
},
|
| 8 |
{
|
| 9 |
"generations": ["(12.7, 67.8)", "(16.7, 89.6)"],
|
| 10 |
"golds": [[12.7, 67.8], [10.9, 80.6]],
|
| 11 |
-
"
|
| 12 |
"result": {"coord_accuracy": 1}
|
| 13 |
},
|
| 14 |
]
|
|
|
|
| 2 |
{
|
| 3 |
"generations": ["(12.7, 67.8)", "(16.7, 89.6)"],
|
| 4 |
"golds": [[12.7, 67.8], [10.9, 80.6]],
|
| 5 |
+
"d_range": 20,
|
| 6 |
"result": {"coord_accuracy": 0.5}
|
| 7 |
},
|
| 8 |
{
|
| 9 |
"generations": ["(12.7, 67.8)", "(16.7, 89.6)"],
|
| 10 |
"golds": [[12.7, 67.8], [10.9, 80.6]],
|
| 11 |
+
"d_range": 1000000,
|
| 12 |
"result": {"coord_accuracy": 1}
|
| 13 |
},
|
| 14 |
]
|