update

ranking_loss.py

@@ -41,7 +41,7 @@ Args:
     references: list of reference for each prediction. Each
         reference should be a string with tokens separated by spaces.
 Returns:
-    accuracy: description of the first score,
+    ranking_loss: description of the first score,
     another_score: description of the second score,
 Examples:
     Examples should be written in doctest format, and should illustrate how
@@ -50,7 +50,7 @@ Examples:
     >>> my_new_module = evaluate.load("my_new_module")
     >>> results = my_new_module.compute(references=[0, 1], predictions=[0, 1])
     >>> print(results)
-    {'accuracy': 1.0}
+    {'ranking_loss': 1.0}
 """
 
 # TODO: Define external resources urls if needed
@@ -71,8 +71,8 @@ class RankingLoss(evaluate.Metric):
             inputs_description=_KWARGS_DESCRIPTION,
             # This defines the format of each prediction and reference
             features=datasets.Features({
-                'predictions': datasets.Value('int64'),
-                'references': datasets.Value('int64'),
+                'predictions': datasets.Sequence('float32'),
+                'references': datasets.Sequence('int64'),
             }),
             # Homepage of the module for documentation
             homepage="http://module.homepage",