Commit a3eb6393 authored by lintangsutawika's avatar lintangsutawika
Browse files

fixed group to int

parent caac0843
......@@ -39,11 +39,10 @@ def em(predictions, references): # This is a passthrough function
else not bool(reference)
)
return (_group, prediction, reference)
return (int(_group), prediction, reference)
def agg_em(items):
grouped_values = collections.defaultdict(lambda: ([], []))
for group, prediction, reference in items:
grouped_values[group][0].append(reference)
......
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