gagan3012 commited on
Commit
6aa8328
·
1 Parent(s): 3e20595
Files changed (1) hide show
  1. src/models/model.py +2 -3
src/models/model.py CHANGED
@@ -482,7 +482,7 @@ class Summarization:
482
  print(type(predictions), type(references))
483
 
484
  results = metric.compute(predictions=predictions, references=references)
485
- '''
486
  output = {
487
  'Rouge 1': {
488
  'Rouge_1 Low Precision': results["rouge1"].low.precision,
@@ -529,5 +529,4 @@ class Summarization:
529
  'rougeLsum High F1': results["rougeLsum"].high.fmeasure,
530
  }
531
  }
532
- '''
533
- return results
 
482
  print(type(predictions), type(references))
483
 
484
  results = metric.compute(predictions=predictions, references=references)
485
+
486
  output = {
487
  'Rouge 1': {
488
  'Rouge_1 Low Precision': results["rouge1"].low.precision,
 
529
  'rougeLsum High F1': results["rougeLsum"].high.fmeasure,
530
  }
531
  }
532
+ return output