Spaces:
Runtime error
Runtime error
arthurvqin
commited on
Commit
•
012bae0
1
Parent(s):
175817c
Modified pr_auc.py
Browse files
pr_auc.py
CHANGED
@@ -77,8 +77,8 @@ class PRAUC(evaluate.Metric):
|
|
77 |
inputs_description=_KWARGS_DESCRIPTION,
|
78 |
features=datasets.Features(
|
79 |
{
|
80 |
-
"prediction_scores": datasets.Sequence(datasets.Value("
|
81 |
-
"references": datasets.Value("
|
82 |
}
|
83 |
),
|
84 |
reference_urls=["https://scikit-learn.org/stable/modules/generated/sklearn.metrics.average_precision_score.html"],
|
|
|
77 |
inputs_description=_KWARGS_DESCRIPTION,
|
78 |
features=datasets.Features(
|
79 |
{
|
80 |
+
"prediction_scores": datasets.Sequence(datasets.Sequence(datasets.Value("uint16"))),
|
81 |
+
"references": datasets.Sequence(datasets.Sequence(datasets.Value("uint16"))),
|
82 |
}
|
83 |
),
|
84 |
reference_urls=["https://scikit-learn.org/stable/modules/generated/sklearn.metrics.average_precision_score.html"],
|
tests.py
DELETED
@@ -1,17 +0,0 @@
|
|
1 |
-
test_cases = [
|
2 |
-
{
|
3 |
-
"predictions": [0, 0],
|
4 |
-
"references": [1, 1],
|
5 |
-
"result": {"metric_score": 0}
|
6 |
-
},
|
7 |
-
{
|
8 |
-
"predictions": [1, 1],
|
9 |
-
"references": [1, 1],
|
10 |
-
"result": {"metric_score": 1}
|
11 |
-
},
|
12 |
-
{
|
13 |
-
"predictions": [1, 0],
|
14 |
-
"references": [1, 1],
|
15 |
-
"result": {"metric_score": 0.5}
|
16 |
-
}
|
17 |
-
]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|