Aye10032 commited on
Commit
81704c2
·
1 Parent(s): b400989
Files changed (1) hide show
  1. 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(list[datasets.Value("float32")]),
48
  "references": datasets.Sequence(datasets.Value("int32")),
49
  }
50
  if self.config_name == "multilabel"
51
  else {
52
- "predictions": list[datasets.Value("float32")],
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
  ),