riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 Packages Projects Releases Wiki Activity

[s2s] consistent output format across eval scripts (#7435)

Browse Source
This commit is contained in:
Sam Shleifer 2020-09-28 23:20:03 -04:00 committed by GitHub
parent 671b278e25
commit 74d8d69bd4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/seq2seq/rouge_cli.py
View File

@ -9,7 +9,7 @@ def calculate_rouge_path(pred_path, tgt_path, save_path=None, **kwargs):
tgt_lns = [x.strip() for x in open(tgt_path).readlines()][: len(pred_lns)]
metrics = calculate_rouge(pred_lns, tgt_lns, **kwargs)
if save_path is not None:
save_json(metrics, save_path)
save_json(metrics, save_path, indent=None)
return metrics # these print nicely

3
examples/seq2seq/run_eval.py
View File

@ -152,8 +152,7 @@ def run_generate(verbose=True):
print(scores)
if args.score_path is not None:
path = args.score_path
json.dump(scores, open(path, "w"))
json.dump(scores, open(args.score_path, "w"))
return scores