Gagan Bhatia commited on
Commit
ab326e4
·
1 Parent(s): d907ff7

Update model.py

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