SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 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
2 changed files with 2 additions and 3 deletions
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)] tgt_lns = [x.strip() for x in open(tgt_path).readlines()][: len(pred_lns)]
metrics = calculate_rouge(pred_lns, tgt_lns, **kwargs) metrics = calculate_rouge(pred_lns, tgt_lns, **kwargs)
if save_path is not None: if save_path is not None:
save_json(metrics, save_path) save_json(metrics, save_path, indent=None)
return metrics # these print nicely return metrics # these print nicely

3
examples/seq2seq/run_eval.py
View File

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