Gagan Bhatia commited on
Commit
be26bf4
1 Parent(s): 6c8b15e

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +5 -0
src/models/model.py CHANGED
@@ -480,3 +480,8 @@ class Summarization:
480
  'Rouge_2 Low Precision': results["rouge2"].low.precision,
481
  'Rouge_2 Low recall': results["rouge2"].low.recall,
482
  'Rouge_2 Low F1': results["rouge2"].low.fmeasure,
 
 
 
 
 
 
480
  'Rouge_2 Low Precision': results["rouge2"].low.precision,
481
  'Rouge_2 Low recall': results["rouge2"].low.recall,
482
  'Rouge_2 Low F1': results["rouge2"].low.fmeasure,
483
+ 'Rouge_2 Mid Precision': results["rouge2"].mid.precision,
484
+ 'Rouge_2 Mid recall': results["rouge2"].mid.recall,
485
+ 'Rouge_2 Mid F1': results["rouge2"].mid.fmeasure,
486
+ 'Rouge_2 High Precision': results["rouge2"].high.precision,
487
+ 'Rouge_2 High recall': results["rouge2"].high.recall,