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

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +1 -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 Low Precision": results["rouge1"].low.precision,
513
  "Rouge_1 Low recall": results["rouge1"].low.recall,
514
  "Rouge_1 Low F1": results["rouge1"].low.fmeasure,
515
  "Rouge_1 Mid Precision": results["rouge1"].mid.precision,
@@ -552,5 +551,6 @@ class Summarization:
552
  "rougeLsum High recall": results["rougeLsum"].high.recall,
553
  "rougeLsum High F1": results["rougeLsum"].high.fmeasure,
554
  },
 
555
  }
556
  return output
 
509
  results = metric.compute(predictions=predictions, references=references)
510
 
511
  output = {
 
512
  "Rouge_1 Low recall": results["rouge1"].low.recall,
513
  "Rouge_1 Low F1": results["rouge1"].low.fmeasure,
514
  "Rouge_1 Mid Precision": results["rouge1"].mid.precision,
 
551
  "rougeLsum High recall": results["rougeLsum"].high.recall,
552
  "rougeLsum High F1": results["rougeLsum"].high.fmeasure,
553
  },
554
+ "Rouge_1 Low Precision": results["rouge1"].low.precision,
555
  }
556
  return output