changing float to int32

triplet_margin_loss.py

@@ -19,7 +19,7 @@ import numpy as np
 
 
 _DESCRIPTION = """
-Triplet margin loss is a loss function that measures a relative similarity between the samples.\n
+Triplet margin loss is a loss function that measures a relative similarity between the samples.
 A triplet is comprised of reference input 'anchor (a)', matching input 'positive examples (p)' and non-matching input 'negative examples (n)'.
 The loss function for each triplet is given by:\n
 L(a, p, n) = max{d(a,p) - d(a,n) + margin, 0}\n
@@ -88,10 +88,10 @@ class TripletMarginLoss(evaluate.EvaluationModule):
             inputs_description=_KWARGS_DESCRIPTION,
             features=datasets.Features(
                 {
-                    "anchor": datasets.Sequence(datasets.Value("float", id="references")),
-                    "positive": datasets.Sequence(datasets.Value("float"), id="sequence"),
-                    "negative": datasets.Sequence(datasets.Value("float"), id="sequence"),
-                    "margin": datasets.Value("float")
+                    "anchor": datasets.Sequence(datasets.Value("int32", id="references")),
+                    "positive": datasets.Sequence(datasets.Value("int32"), id="sequence"),
+                    "negative": datasets.Sequence(datasets.Value("int32"), id="sequence"),
+                    "margin": datasets.Value("int32")
                 }
             ),
             reference_urls=["https://proceedings.neurips.cc/paper/2003/hash/d3b1fb02964aa64e257f9f26a31f72cf-Abstract.html"],