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

Mult rouge by 100: standard units (#6359)

Browse Source
This commit is contained in:
Sam Shleifer
2020-08-13 12:15:54 -04:00
committed by GitHub
parent eda07efaa5
commit e92efcf728
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/seq2seq/utils.py
View File

@@ -271,7 +271,7 @@ def calculate_rouge(output_lns: List[str], reference_lns: List[str], use_stemmer
aggregator.add_scores(scores) aggregator.add_scores(scores)
result = aggregator.aggregate() 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): def freeze_params(model: nn.Module):