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

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +3 -0
src/models/model.py CHANGED
@@ -513,5 +513,8 @@ class Summarization:
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,
 
 
 
516
  }
517
  return output
 
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,
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