From 5187bfb4c67cfc89a649e8d068dc9a9760012c0a Mon Sep 17 00:00:00 2001 From: Oleg Date: Sun, 16 Apr 2023 11:01:50 +0500 Subject: [PATCH] fix _take_tails --- implicit/evaluation.pyx | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/implicit/evaluation.pyx b/implicit/evaluation.pyx index f25d0da..976b0c7 100644 --- a/implicit/evaluation.pyx +++ b/implicit/evaluation.pyx @@ -122,7 +122,8 @@ cdef _take_tails(arr, int n, return_complement=False, shuffled=False): idx = arr.argsort() sorted_arr = arr[idx] - end = np.bincount(sorted_arr).cumsum() - 1 + _, counts = np.unique(sorted_arr, return_counts=True) + end = counts.cumsum() - 1 start = end - n ranges = np.linspace(start, end, num=n + 1, dtype=int)[1:]