Gagan Bhatia commited on
Commit
2981c19
·
1 Parent(s): ab326e4

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +3 -0
src/models/model.py CHANGED
@@ -510,5 +510,8 @@ class Summarization:
510
 
511
  output = {
512
  "Rouge_1 Low Precision": results["rouge1"].low.precision,
 
 
 
513
  }
514
  return output
 
510
 
511
  output = {
512
  "Rouge_1 Low Precision": results["rouge1"].low.precision,
513
+ "Rouge_1 Low recall": results["rouge1"].low.recall,
514
+ "Rouge_1 Low F1": results["rouge1"].low.fmeasure,
515
+ "Rouge_1 Mid Precision": results["rouge1"].mid.precision,
516
  }
517
  return output