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,
|