cifar10

keras-team · May 14, 2018 · 5858cf3 · 5858cf3
1 parent 16a8e4a
commit 5858cf3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/autokeras/search.py b/autokeras/search.py
@@ -219,8 +219,8 @@ def search(self, x_train, y_train, x_test, y_test):
 
             graph = Graph(model)
             self.init_search_queue = []
-            for child_graph in transform(graph):
-                self.init_search_queue.append((child_graph, history_item['model_id']))
+            # for child_graph in transform(graph):
+            #     self.init_search_queue.append((child_graph, history_item['model_id']))
             self.init_gpr_x.append(graph.extract_descriptor())
             self.init_gpr_y.append(history_item['accuracy'])
             pickle_to_file(self, os.path.join(self.path, 'searcher'))

diff --git a/experiments/cifar10_2.py b/experiments/cifar10_2.py
@@ -28,7 +28,8 @@ def select_gpu():
     X = np.concatenate((x_train, x_test))
     Y = np.concatenate((y_train, y_test))
     trainer_args = {'max_iter_num': 0}
-    clf = ImageClassifier(searcher_type='bayesian', path='/Users/haifeng/cifar102', verbose=True, trainer_args=trainer_args)
+    clf = ImageClassifier(searcher_type='bayesian', path='/Users/haifeng/cifar102', verbose=True,
+                          searcher_args={'trainer_args': {'max_iter_num': 0}, 'default_model_len': 10})
 
     clf.fit(x_train, y_train, time_limit=12*60*60)
     # clf.final_fit(x_train, y_train)