fix input format
Browse files
ece.py
CHANGED
@@ -76,7 +76,7 @@ class ECE(evaluate.Metric):
|
|
76 |
inputs_description=_KWARGS_DESCRIPTION,
|
77 |
# This defines the format of each prediction and reference
|
78 |
features=datasets.Features({
|
79 |
-
'predictions': datasets.Value('float32'),
|
80 |
'references': datasets.Value('int64'),
|
81 |
}),
|
82 |
# Homepage of the module for documentation
|
|
|
76 |
inputs_description=_KWARGS_DESCRIPTION,
|
77 |
# This defines the format of each prediction and reference
|
78 |
features=datasets.Features({
|
79 |
+
'predictions': datasets.Sequence(datasets.Value('float32')),
|
80 |
'references': datasets.Value('int64'),
|
81 |
}),
|
82 |
# Homepage of the module for documentation
|