Spaces:
Sleeping
Sleeping
remove ids
Browse files- triplet_margin_loss.py +3 -3
triplet_margin_loss.py
CHANGED
@@ -88,9 +88,9 @@ class TripletMarginLoss(evaluate.EvaluationModule):
|
|
88 |
inputs_description=_KWARGS_DESCRIPTION,
|
89 |
features=datasets.Features(
|
90 |
{
|
91 |
-
"anchor": datasets.Sequence(datasets.Value("float"
|
92 |
-
"positive": datasets.Sequence(datasets.Value("float")
|
93 |
-
"negative": datasets.Sequence(datasets.Value("float")
|
94 |
"margin": datasets.Value("float")
|
95 |
}
|
96 |
),
|
|
|
88 |
inputs_description=_KWARGS_DESCRIPTION,
|
89 |
features=datasets.Features(
|
90 |
{
|
91 |
+
"anchor": datasets.Sequence(datasets.Value("float")),
|
92 |
+
"positive": datasets.Sequence(datasets.Value("float")),
|
93 |
+
"negative": datasets.Sequence(datasets.Value("float")),
|
94 |
"margin": datasets.Value("float")
|
95 |
}
|
96 |
),
|