From e92efcf7286c955e6901f894be39cf6154af48b7 Mon Sep 17 00:00:00 2001 From: Sam Shleifer Date: Thu, 13 Aug 2020 12:15:54 -0400 Subject: [PATCH] Mult rouge by 100: standard units (#6359) --- examples/seq2seq/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/seq2seq/utils.py b/examples/seq2seq/utils.py index 8c8b3005c4..b96158f21a 100644 --- a/examples/seq2seq/utils.py +++ b/examples/seq2seq/utils.py @@ -271,7 +271,7 @@ def calculate_rouge(output_lns: List[str], reference_lns: List[str], use_stemmer aggregator.add_scores(scores) result = aggregator.aggregate() - return {k: v.mid.fmeasure for k, v in result.items()} + return {k: v.mid.fmeasure * 100 for k, v in result.items()} def freeze_params(model: nn.Module):