Vlasta commited on
Commit
acd2531
1 Parent(s): d406c9d

Fix float input

Browse files
Files changed (1) hide show
  1. pr_auc.py +1 -1
pr_auc.py CHANGED
@@ -72,7 +72,7 @@ class PRAUC(evaluate.Metric):
72
  inputs_description=_KWARGS_DESCRIPTION,
73
  # This defines the format of each prediction and reference
74
  features=datasets.Features({
75
- 'predictions': datasets.Value('int64'),
76
  'references': datasets.Value('int64'),
77
  }),
78
  # Homepage of the module for documentation
 
72
  inputs_description=_KWARGS_DESCRIPTION,
73
  # This defines the format of each prediction and reference
74
  features=datasets.Features({
75
+ 'prediction_scores': datasets.Sequence(datasets.Value("float")),
76
  'references': datasets.Value('int64'),
77
  }),
78
  # Homepage of the module for documentation