Added cached hash into for loop.

maheshakya · Jul 20, 2014 · 8470bad · 8470bad
1 parent cde73a4
commit 8470bad
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/sklearn/neighbors/lsh_forest.py b/sklearn/neighbors/lsh_forest.py
@@ -258,8 +258,11 @@ def fit(self, X=None):
         self.cache_N = int(self.max_label_length/2)
         hashes = [x for x in itertools.product((0, 1),
                                                repeat=self.cache_N)]
-        self.cache = {tuple(x): int("".join([digits[y] for y in x]),
-                                    2) for x in hashes}
+
+        self.cache = {}
+        for item in hashes:
+            self.cache[tuple(item)] = int("".join([digits[y] for y in item]),
+                                          2)
 
         self.k = 2 ** self.cache_N