Commit 1a153185 authored by Konstantin Schulz's avatar Konstantin Schulz
Browse files

fix key access in squad evaluation metrics

parent 4645584f
...@@ -40,7 +40,7 @@ def _squad_metric(predictions, references): ...@@ -40,7 +40,7 @@ def _squad_metric(predictions, references):
def _squad_agg(key, items): def _squad_agg(key, items):
predictions, references = zip(*items) predictions, references = zip(*items)
return _squad_metric(predictions=predictions, references=references)[key] return _squad_metric(predictions=predictions, references=references).get(key, 0)
class SQuAD2(Task): class SQuAD2(Task):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment