omidf commited on
Commit
1fa4eab
·
1 Parent(s): a86f57e

Update compute_score.py

Browse files
Files changed (1) hide show
  1. compute_score.py +2 -1
compute_score.py CHANGED
@@ -85,7 +85,8 @@ def compute_score(dataset, predictions):
85
  prediction = predictions[qa["id"]]
86
  exact_match += metric_max_over_ground_truths(exact_match_score, prediction, ground_truths)
87
  precision_qa = metric_max_over_ground_truths(precision_score, prediction, ground_truths)
88
- recall_qa = metric_max_over_ground_truths(recall_score, prediction, ground_truths)
 
89
  f1 += (2 * precision_qa * recall_qa) / (precision_qa + recall_qa)
90
  recall += recall_qa
91
  precision += precision_qa
 
85
  prediction = predictions[qa["id"]]
86
  exact_match += metric_max_over_ground_truths(exact_match_score, prediction, ground_truths)
87
  precision_qa = metric_max_over_ground_truths(precision_score, prediction, ground_truths)
88
+ recall_qa = metric_max_over_ground_truths(recall_score, prediction, ground_truths)\
89
+ print(recall_qa, precision_qa)
90
  f1 += (2 * precision_qa * recall_qa) / (precision_qa + recall_qa)
91
  recall += recall_qa
92
  precision += precision_qa