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

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +4 -0
src/models/model.py CHANGED
@@ -522,5 +522,9 @@ class Summarization:
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
 
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
+ "Rouge_2 Mid recall": results["rouge2"].mid.recall,
526
+ "Rouge_2 Mid F1": results["rouge2"].mid.fmeasure,
527
+ "Rouge_2 High Precision": results["rouge2"].high.precision,
528
+ "Rouge_2 High recall": results["rouge2"].high.recall,
529
  }
530
  return output