Rodrigo Ferreira Rodrigues commited on
Commit
1460313
·
1 Parent(s): 1b88c4f

Adding ast library

Browse files
Files changed (2) hide show
  1. coord_eval.py +3 -2
  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]], d=20)
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 += accuracy_coord(f_ans, f_gold, d_range)
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
- "d": 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": 1000000,
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
  ]