Gagan Bhatia commited on
Commit
20feb06
1 Parent(s): 589ca92

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +6 -0
src/models/model.py CHANGED
@@ -497,3 +497,9 @@ class Summarization:
497
  'Rouge_L High Precision': results["rougeL"].high.precision,
498
  'Rouge_L High recall': results["rougeL"].high.recall,
499
  'Rouge_L High F1': results["rougeL"].high.fmeasure,
 
 
 
 
 
 
 
497
  'Rouge_L High Precision': results["rougeL"].high.precision,
498
  'Rouge_L High recall': results["rougeL"].high.recall,
499
  'Rouge_L High F1': results["rougeL"].high.fmeasure,
500
+ },
501
+ 'rougeLsum': {
502
+ 'rougeLsum Low Precision': results["rougeLsum"].low.precision,
503
+ 'rougeLsum Low recall': results["rougeLsum"].low.recall,
504
+ 'rougeLsum Low F1': results["rougeLsum"].low.fmeasure,
505
+ 'rougeLsum Mid Precision': results["rougeLsum"].mid.precision,