fix top_k for toppop (#73)

smartnews · Sep 5, 2022 · 6a9f108 · 6a9f108
1 parent eefdf5f
commit 6a9f108
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/rsdiv/recommenders/ials.py b/src/rsdiv/recommenders/ials.py
@@ -147,7 +147,7 @@ def get_topk_single_user(
         """
         scores = self.get_score_single_user(user_string, keep_indices)
         if scores is None:
-            original_indices = self._get_toppop_keep(keep_indices)
+            original_indices = self._get_toppop_keep(keep_indices)[:top_k]
         else:
             rank = self.get_topk_indices(scores, top_k)
             original_indices = keep_indices[rank]