dnrops
/
transformers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats 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)
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):