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

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +4 -0
src/models/model.py CHANGED
@@ -503,3 +503,7 @@ class Summarization:
503
  'rougeLsum Low recall': results["rougeLsum"].low.recall,
504
  'rougeLsum Low F1': results["rougeLsum"].low.fmeasure,
505
  'rougeLsum Mid Precision': results["rougeLsum"].mid.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,
506
+ 'rougeLsum Mid recall': results["rougeLsum"].mid.recall,
507
+ 'rougeLsum Mid F1': results["rougeLsum"].mid.fmeasure,
508
+ 'rougeLsum High Precision': results["rougeLsum"].high.precision,
509
+ 'rougeLsum High recall': results["rougeLsum"].high.recall,