update

README.md

@@ -1,7 +1,7 @@
 ---
 title: Ranking Loss
 datasets:
--  
+- "custom"
 tags:
 - evaluate
 - metric

ranking_loss.py

@@ -15,7 +15,7 @@
 
 import evaluate
 import datasets
-
+from sklearn.metrics import label_ranking_loss
 
 # TODO: Add BibTeX citation
 _CITATION = """\
@@ -89,7 +89,6 @@ class RankingLoss(evaluate.Metric):
     def _compute(self, predictions, references):
         """Returns the scores"""
         # TODO: Compute the different scores of the module
-        accuracy = sum(i == j for i, j in zip(predictions, references)) / len(predictions)
         return {
-            "accuracy": accuracy,
+            "ranking_loss": label_ranking_loss(references, predictions),
         }

requirements.txt

@@ -1 +1,2 @@
-git+https://github.com/huggingface/evaluate@main
+git+https://github.com/huggingface/evaluate@main
+scikit-learn