Gagan Bhatia commited on
Commit
ac42dfa
1 Parent(s): cf02d2a

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +3 -0
src/models/model.py CHANGED
@@ -519,5 +519,8 @@ class Summarization:
519
  "Rouge_1 High recall": results["rouge1"].high.recall,
520
  "Rouge_1 High F1": results["rouge1"].high.fmeasure,
521
  "Rouge_2 Low Precision": results["rouge2"].low.precision,
 
 
 
522
  }
523
  return output
 
519
  "Rouge_1 High recall": results["rouge1"].high.recall,
520
  "Rouge_1 High F1": results["rouge1"].high.fmeasure,
521
  "Rouge_2 Low Precision": results["rouge2"].low.precision,
522
+ "Rouge_2 Low recall": results["rouge2"].low.recall,
523
+ "Rouge_2 Low F1": results["rouge2"].low.fmeasure,
524
+ "Rouge_2 Mid Precision": results["rouge2"].mid.precision,
525
  }
526
  return output