Update metric_template_1.py: Adjusted features structure for multilabel config
Browse files- metric_template_1.py +2 -2
metric_template_1.py
CHANGED
|
@@ -94,13 +94,13 @@ class ISCO_Hierarchical_Accuracy(evaluate.Metric):
|
|
| 94 |
# ),
|
| 95 |
features=datasets.Features(
|
| 96 |
{
|
| 97 |
-
"predictions": datasets.Sequence(datasets.Value("string")),
|
| 98 |
"references": datasets.Sequence(datasets.Value("string")),
|
|
|
|
| 99 |
}
|
| 100 |
if self.config_name == "multilabel"
|
| 101 |
else {
|
| 102 |
-
"predictions": datasets.Value("string"),
|
| 103 |
"references": datasets.Value("string"),
|
|
|
|
| 104 |
}
|
| 105 |
),
|
| 106 |
# TODO: Homepage of the module for documentation
|
|
|
|
| 94 |
# ),
|
| 95 |
features=datasets.Features(
|
| 96 |
{
|
|
|
|
| 97 |
"references": datasets.Sequence(datasets.Value("string")),
|
| 98 |
+
"predictions": datasets.Sequence(datasets.Value("string")),
|
| 99 |
}
|
| 100 |
if self.config_name == "multilabel"
|
| 101 |
else {
|
|
|
|
| 102 |
"references": datasets.Value("string"),
|
| 103 |
+
"predictions": datasets.Value("string"),
|
| 104 |
}
|
| 105 |
),
|
| 106 |
# TODO: Homepage of the module for documentation
|