Mult rouge by 100: standard units (#6359)

This commit is contained in:
Sam Shleifer
2020-08-13 12:15:54 -04:00
committed by GitHub
parent eda07efaa5
commit e92efcf728

View File

@@ -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):