Gagan Bhatia commited on
Commit
d513beb
1 Parent(s): cd07047

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +3 -0
src/models/model.py CHANGED
@@ -530,5 +530,8 @@ class Summarization:
530
  "Rouge_L Low Precision": results["rougeL"].low.precision,
531
  "Rouge_L Low recall": results["rougeL"].low.recall,
532
  "Rouge_L Low F1": results["rougeL"].low.fmeasure,
 
 
 
533
  }
534
  return output
 
530
  "Rouge_L Low Precision": results["rougeL"].low.precision,
531
  "Rouge_L Low recall": results["rougeL"].low.recall,
532
  "Rouge_L Low F1": results["rougeL"].low.fmeasure,
533
+ "Rouge_L Mid Precision": results["rougeL"].mid.precision,
534
+ "Rouge_L Mid recall": results["rougeL"].mid.recall,
535
+ "Rouge_L Mid F1": results["rougeL"].mid.fmeasure,
536
  }
537
  return output