Gagan Bhatia commited on
Commit
6c6d23b
·
1 Parent(s): fc0c429

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +0 -1
src/models/model.py CHANGED
@@ -509,7 +509,6 @@ class Summarization:
509
  results = metric.compute(predictions=predictions, references=references)
510
 
511
  output = {
512
- "Rouge 1": {
513
  "Rouge_1 Low Precision": results["rouge1"].low.precision,
514
  "Rouge_1 Low recall": results["rouge1"].low.recall,
515
  "Rouge_1 Low F1": results["rouge1"].low.fmeasure,
 
509
  results = metric.compute(predictions=predictions, references=references)
510
 
511
  output = {
 
512
  "Rouge_1 Low Precision": results["rouge1"].low.precision,
513
  "Rouge_1 Low recall": results["rouge1"].low.recall,
514
  "Rouge_1 Low F1": results["rouge1"].low.fmeasure,