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

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +6 -0
src/models/model.py CHANGED
@@ -474,3 +474,9 @@ class Summarization:
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,
 
 
 
 
 
 
 
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,
477
+ 'Rouge_1 High F1': results["rouge1"].high.fmeasure,
478
+ },
479
+ 'Rouge 2': {
480
+ 'Rouge_2 Low Precision': results["rouge2"].low.precision,
481
+ 'Rouge_2 Low recall': results["rouge2"].low.recall,
482
+ 'Rouge_2 Low F1': results["rouge2"].low.fmeasure,