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