arthurvqin commited on
Commit
012bae0
1 Parent(s): 175817c

Modified pr_auc.py

Browse files
Files changed (2) hide show
  1. pr_auc.py +2 -2
  2. tests.py +0 -17
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("float")),
81
- "references": datasets.Value("int32"),
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
- ]