Gagan Bhatia commited on
Commit
80331f7
1 Parent(s): be26bf4

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +6 -0
src/models/model.py CHANGED
@@ -485,3 +485,9 @@ class Summarization:
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,
 
 
 
 
 
 
 
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,
488
+ 'Rouge_2 High F1': results["rouge2"].high.fmeasure,
489
+ },
490
+ 'Rouge L': {
491
+ 'Rouge_L Low Precision': results["rougeL"].low.precision,
492
+ 'Rouge_L Low recall': results["rougeL"].low.recall,
493
+ 'Rouge_L Low F1': results["rougeL"].low.fmeasure,