bug fixed (#267)

keras-team · Oct 19, 2018 · d5f3b36 · d5f3b36
1 parent fbc7987
commit d5f3b36
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/autokeras/image/image_supervised.py b/autokeras/image/image_supervised.py
@@ -158,9 +158,11 @@ def fit(self, x, y, x_test=None, y_test=None, time_limit=None):
         y = self.transform_y(y)
         if x_test is None or y_test is None:
             # Divide training data into training and testing data.
+            validation_set_size = int(len(y) * Constant.VALIDATION_SET_SIZE)
+            validation_set_size = min(validation_set_size, 500)
+            validation_set_size = max(validation_set_size, 1)
             x_train, x_test, y_train, y_test = train_test_split(x, y,
-                                                                test_size=min(Constant.VALIDATION_SET_SIZE,
-                                                                              int(len(y) * 0.2)),
+                                                                test_size=validation_set_size,
                                                                 random_state=42)
         else:
             x_train = x

diff --git a/tests/image/test_image_supervised.py b/tests/image/test_image_supervised.py
@@ -74,7 +74,7 @@ def test_timeout_resume(_):
     # make it impossible to complete within 10sec
     Constant.MAX_MODEL_NUM = 1000
     Constant.SEARCH_MAX_ITER = 1
-    Constant.T_MIN = 0.8
+    # Constant.T_MIN = 0.8
     train_x = np.random.rand(100, 25, 25, 1)
     train_y = np.random.randint(0, 5, 100)
     test_x = np.random.rand(100, 25, 25, 1)