Spaces:
Sleeping
Sleeping
Update compute_score.py
Browse files- compute_score.py +3 -2
compute_score.py
CHANGED
|
@@ -86,8 +86,9 @@ def compute_score(dataset, predictions):
|
|
| 86 |
exact_match += metric_max_over_ground_truths(exact_match_score, prediction, ground_truths)
|
| 87 |
recall_temp = metric_max_over_ground_truths(recall_score, prediction, ground_truths)
|
| 88 |
precision_temp = metric_max_over_ground_truths(precision_score, prediction, ground_truths)
|
| 89 |
-
|
| 90 |
-
|
|
|
|
| 91 |
f1_temp = 0
|
| 92 |
else:
|
| 93 |
f1_temp = (2 * precision * recall) / (precision + recall)
|
|
|
|
| 86 |
exact_match += metric_max_over_ground_truths(exact_match_score, prediction, ground_truths)
|
| 87 |
recall_temp = metric_max_over_ground_truths(recall_score, prediction, ground_truths)
|
| 88 |
precision_temp = metric_max_over_ground_truths(precision_score, prediction, ground_truths)
|
| 89 |
+
|
| 90 |
+
print(recall_temp + precision_temp)
|
| 91 |
+
if recall_temp + precision_temp == 0:
|
| 92 |
f1_temp = 0
|
| 93 |
else:
|
| 94 |
f1_temp = (2 * precision * recall) / (precision + recall)
|