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

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +12 -0
src/models/model.py CHANGED
@@ -533,5 +533,17 @@ class Summarization:
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
 
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
+ "Rouge_L High Precision": results["rougeL"].high.precision,
537
+ "Rouge_L High recall": results["rougeL"].high.recall,
538
+ "Rouge_L High F1": results["rougeL"].high.fmeasure,
539
+ "rougeLsum Low Precision": results["rougeLsum"].low.precision,
540
+ "rougeLsum Low recall": results["rougeLsum"].low.recall,
541
+ "rougeLsum Low F1": results["rougeLsum"].low.fmeasure,
542
+ "rougeLsum Mid Precision": results["rougeLsum"].mid.precision,
543
+ "rougeLsum Mid recall": results["rougeLsum"].mid.recall,
544
+ "rougeLsum Mid F1": results["rougeLsum"].mid.fmeasure,
545
+ "rougeLsum High Precision": results["rougeLsum"].high.precision,
546
+ "rougeLsum High recall": results["rougeLsum"].high.recall,
547
+ "rougeLsum High F1": results["rougeLsum"].high.fmeasure,
548
  }
549
  return output