keys

keras-team · May 17, 2018 · 2c573ea · 2c573ea
1 parent ff0291e
commit 2c573ea
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/autokeras/search.py b/autokeras/search.py
@@ -250,7 +250,7 @@ def search(self, x_train, y_train, x_test, y_test):
         pickle_to_file(self, os.path.join(self.path, 'searcher'))
 
     def maximize_acq(self):
-        model_ids = self.search_tree.get_leaves()
+        model_ids = self.search_tree.adj_list.keys()
         target_graph = None
         father_id = None
         descriptors = self.descriptors