From 94b372898924b1e1365eda8320164f3ec7817f5c Mon Sep 17 00:00:00 2001 From: Marco Vela <36453977+characat0@users.noreply.github.com> Date: Mon, 27 May 2024 12:02:41 -0500 Subject: [PATCH] Update evaluation.pyx doctoring --- implicit/evaluation.pyx | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/implicit/evaluation.pyx b/implicit/evaluation.pyx index bd590401..4ce6dd0c 100644 --- 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()