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

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +3 -0
src/models/model.py CHANGED
@@ -516,5 +516,8 @@ class Summarization:
516
  "Rouge_1 Mid recall": results["rouge1"].mid.recall,
517
  "Rouge_1 Mid F1": results["rouge1"].mid.fmeasure,
518
  "Rouge_1 High Precision": results["rouge1"].high.precision,
 
 
 
519
  }
520
  return output
 
516
  "Rouge_1 Mid recall": results["rouge1"].mid.recall,
517
  "Rouge_1 Mid F1": results["rouge1"].mid.fmeasure,
518
  "Rouge_1 High Precision": results["rouge1"].high.precision,
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