Spaces:
Sleeping
Sleeping
update
Browse files- top5_error_rate.py +2 -2
top5_error_rate.py
CHANGED
@@ -44,12 +44,12 @@ class Top5ErrorRate(evaluate.Metric):
|
|
44 |
inputs_description=_KWARGS_DESCRIPTION,
|
45 |
features=datasets.Features(
|
46 |
{
|
47 |
-
"predictions": datasets.Sequence(
|
48 |
"references": datasets.Sequence(datasets.Value("int32")),
|
49 |
}
|
50 |
if self.config_name == "multilabel"
|
51 |
else {
|
52 |
-
"predictions":
|
53 |
"references": datasets.Value("int32"),
|
54 |
}
|
55 |
),
|
|
|
44 |
inputs_description=_KWARGS_DESCRIPTION,
|
45 |
features=datasets.Features(
|
46 |
{
|
47 |
+
"predictions": datasets.Sequence(datasets.Value("float32")),
|
48 |
"references": datasets.Sequence(datasets.Value("int32")),
|
49 |
}
|
50 |
if self.config_name == "multilabel"
|
51 |
else {
|
52 |
+
"predictions": datasets.Value("float32"),
|
53 |
"references": datasets.Value("int32"),
|
54 |
}
|
55 |
),
|