knearest fixed

uclid-org · Oct 13, 2020 · a6a538a · a6a538a
1 parent 1c3a06f
commit a6a538a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/medleysolver/classifiers.py b/medleysolver/classifiers.py
@@ -178,7 +178,9 @@ def get_ordering(self, point, count):
         if np.random.rand() >= self.epsilon * (self.decay ** count) and self.solved:
             candidates = sorted(self.solved, key=lambda entry: np.linalg.norm(entry.datapoint - point))[:self.k]
             methods = [x.solve_method for x in candidates]
-            order = sorted(random.shuffle(list(SOLVERS.keys())), key= lambda x: methods.count(x))
+            ss = list(SOLVERS.keys())
+            random.shuffle(ss)
+            order = sorted(ss, key= lambda x: methods.count(x))
         else:
             order = Random.get_ordering(self, point, count)