Gagan Bhatia commited on
Commit
c1f35ff
1 Parent(s): ab10bc6

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +6 -0
src/models/model.py CHANGED
@@ -468,3 +468,9 @@ class Summarization:
468
  'Rouge 1': {
469
  'Rouge_1 Low Precision': results["rouge1"].low.precision,
470
  'Rouge_1 Low recall': results["rouge1"].low.recall,
 
 
 
 
 
 
 
468
  'Rouge 1': {
469
  'Rouge_1 Low Precision': results["rouge1"].low.precision,
470
  'Rouge_1 Low recall': results["rouge1"].low.recall,
471
+ 'Rouge_1 Low F1': results["rouge1"].low.fmeasure,
472
+ 'Rouge_1 Mid Precision': results["rouge1"].mid.precision,
473
+ 'Rouge_1 Mid recall': results["rouge1"].mid.recall,
474
+ 'Rouge_1 Mid F1': results["rouge1"].mid.fmeasure,
475
+ 'Rouge_1 High Precision': results["rouge1"].high.precision,
476
+ 'Rouge_1 High recall': results["rouge1"].high.recall,