sort by similarity only in KNNBasic (as before)

NicolasHug · Sep 29, 2017 · 632fd5c · 632fd5c
1 parent 49c33fa
commit 632fd5c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/surprise/prediction_algorithms/knns.py b/surprise/prediction_algorithms/knns.py
@@ -100,7 +100,7 @@ def estimate(self, u, i):
         x, y = self.switch(u, i)
 
         neighbors = [(self.sim[x, x2], r) for (x2, r) in self.yr[y]]
-        k_neighbors = heapq.nlargest(self.k, neighbors)
+        k_neighbors = heapq.nlargest(self.k, neighbors, key=lambda t: t[0])
 
         # compute weighted average
         sum_sim = sum_ratings = actual_k = 0