Spaces:
Runtime error
Runtime error
Correct return key
Browse files
average_precision_score.py
CHANGED
|
@@ -159,7 +159,7 @@ class AveragePrecision(evaluate.Metric):
|
|
| 159 |
):
|
| 160 |
"""Returns the scores."""
|
| 161 |
return {
|
| 162 |
-
"
|
| 163 |
y_true=references,
|
| 164 |
y_score=prediction_scores,
|
| 165 |
average=average,
|
|
|
|
| 159 |
):
|
| 160 |
"""Returns the scores."""
|
| 161 |
return {
|
| 162 |
+
"average_precision_score": average_precision_score(
|
| 163 |
y_true=references,
|
| 164 |
y_score=prediction_scores,
|
| 165 |
average=average,
|