cpu

keras-team · May 27, 2018 · 7752345 · 7752345
1 parent 7f30b24
commit 7752345
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/autokeras/constant.py b/autokeras/constant.py
@@ -7,8 +7,9 @@
 BETA = 2.576
 KERNEL_LAMBDA = 1.0
 T_MIN = 0.0001
-# T_MIN = 0.8
 N_NEIGHBOURS = 8
+# T_MIN = 0.8
+# N_NEIGHBOURS = 1
 
 # Model Defaults
 

diff --git a/tests/test_search.py b/tests/test_search.py
@@ -32,10 +32,10 @@ def test_bayesian_searcher(_, _1):
     generator = BayesianSearcher(3, (28, 28, 1), verbose=False, path=default_test_path)
     constant.N_NEIGHBOURS = 1
     constant.T_MIN = 0.8
-    for _ in range(4):
+    for _ in range(2):
         generator.search(x_train, y_train, x_test, y_test)
     clean_dir(default_test_path)
-    assert len(generator.history) == 4
+    assert len(generator.history) == 2
 
 
 def test_search_tree():