Fix ROUGE add example check and update README (#18398)

* Fix ROUGE add example check and update README

* Stay consistent in values
This commit is contained in:
Sylvain Gugger
2022-08-01 11:14:49 -04:00
committed by GitHub
parent 62098b9348
commit 941d233153
17 changed files with 60 additions and 28 deletions

View File

@@ -680,10 +680,7 @@ def main():
metric.add_batch(predictions=decoded_preds, references=decoded_labels)
result = metric.compute(use_stemmer=True)
# Extract a few results from ROUGE
result = {key: value.mid.fmeasure * 100 for key, value in result.items()}
result = {k: round(v, 4) for k, v in result.items()}
result = {k: round(v * 100, 4) for k, v in result.items()}
logger.info(result)
# endregion