Commit d75cbc7a authored by Baber's avatar Baber
Browse files

return multiple metrics

parent 5b159bf9
...@@ -1561,7 +1561,10 @@ class ConfigurableTask(Task): ...@@ -1561,7 +1561,10 @@ class ConfigurableTask(Task):
result_score = self._metric_fn_list[metric]([gold, result]) result_score = self._metric_fn_list[metric]([gold, result])
if isinstance(result_score, dict): if isinstance(result_score, dict):
# TODO: this handles the case where HF evaluate returns a dict. # TODO: this handles the case where HF evaluate returns a dict.
result_score = result_score[metric] # result_score = result_score[metric]
for k, v in result_score.items():
result_dict[k] = v
return result_dict
result_dict[metric] = result_score result_dict[metric] = result_score
else: else:
raise ValueError( raise ValueError(
......
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