diff --git a/implicit/evaluation.pyx b/implicit/evaluation.pyx index 00cdf9f0..1975188f 100644 --- a/implicit/evaluation.pyx +++ b/implicit/evaluation.pyx @@ -230,7 +230,7 @@ def ALS_recommend_all(model, users_items, int k=10, int threads=1, show_progress """ Recommends items for all users - Calculates the k best recommendations for all users, and returns numpy ndarray of + Calculates the k best recommendations for all users, and returns numpy ndarray of shape (number_users, k) with scores. Parameters @@ -259,7 +259,7 @@ def ALS_recommend_all(model, users_items, int k=10, int threads=1, show_progress numpy ndarray Array of (number_users, k) with scores """ - + if not isinstance(users_items, csr_matrix): users_items = users_items.tocsr() factors_items = model.item_factors.T