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
|