Update evaluation.pyx doctoring

characat0 · web-flow · commit 94b372898924 · 2024-05-27T12:02:41.000-05:00
diff --git a/implicit/evaluation.pyx b/implicit/evaluation.pyx
@@ -389,8 +389,8 @@ def ranking_metrics_at_k(model, train_user_items, test_user_items, int K=10,
 
     Returns
     -------
-    float
-        the calculated p@k
+    dict[str, float]
+        the calculated metrics@k
     """
     if not isinstance(train_user_items, csr_matrix):
         train_user_items = train_user_items.tocsr()
