SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[s2s] round bleu, rouge to 4 digits (#6704)

Browse Source
This commit is contained in:
Sam Shleifer
2020-08-25 00:33:11 -04:00
committed by GitHub
parent b6512d2357
commit 0344428f79
4 changed files with 12 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/seq2seq/finetune.py
View File

@@ -23,7 +23,7 @@ try:
Seq2SeqDataset,
TranslationDataset,
assert_all_frozen,
calculate_bleu_score,
calculate_bleu,
calculate_rouge,
flatten_list,
freeze_params,
@@ -42,7 +42,7 @@ except ImportError:
Seq2SeqDataset,
TranslationDataset,
assert_all_frozen,
calculate_bleu_score,
calculate_bleu,
calculate_rouge,
flatten_list,
freeze_params,
@@ -325,7 +325,7 @@ class TranslationModule(SummarizationModule):
self.dataset_kwargs["tgt_lang"] = hparams.tgt_lang
def calc_generative_metrics(self, preds, target) -> dict:
return calculate_bleu_score(preds, target)
return calculate_bleu(preds, target)
def main(args, model=None) -> SummarizationModule: