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

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +7 -0
src/models/model.py CHANGED
@@ -461,3 +461,10 @@ class Summarization:
461
  input_text = test_df['input_text']
462
  references = test_df['output_text']
463
  predictions = [self.predict(x) for x in input_text]
 
 
 
 
 
 
 
 
461
  input_text = test_df['input_text']
462
  references = test_df['output_text']
463
  predictions = [self.predict(x) for x in input_text]
464
+
465
+ results = metric.compute(predictions=predictions, references=references)
466
+
467
+ output = {
468
+ 'Rouge 1': {
469
+ 'Rouge_1 Low Precision': results["rouge1"].low.precision,
470
+ 'Rouge_1 Low recall': results["rouge1"].low.recall,