Gagan Bhatia commited on
Commit
a739a9d
·
1 Parent(s): be8895a

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +1 -0
src/models/model.py CHANGED
@@ -507,6 +507,7 @@ class Summarization:
507
  'rougeLsum Mid F1': results["rougeLsum"].mid.fmeasure,
508
  'rougeLsum High Precision': results["rougeLsum"].high.precision,
509
  'rougeLsum High recall': results["rougeLsum"].high.recall,
 
510
  }
511
  }
512
  return output
 
507
  'rougeLsum Mid F1': results["rougeLsum"].mid.fmeasure,
508
  'rougeLsum High Precision': results["rougeLsum"].high.precision,
509
  'rougeLsum High recall': results["rougeLsum"].high.recall,
510
+ 'rougeLsum High F1': results["rougeLsum"].high.fmeasure,
511
  }
512
  }
513
  return output