Testing codepath for splitting data into train/valid sets automatically.

aigamedev · Apr 21, 2015 · bbcae02 · bbcae02
1 parent 69c1a40
commit bbcae02
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/sknn/tests/test_training.py b/sknn/tests/test_training.py
@@ -19,6 +19,15 @@ def test_FitTerminateStable(self):
 
         self.nn._fit(a_in, a_out)
 
+    def test_FitAutomaticValidation(self):
+        a_in, a_out = numpy.zeros((8,16)), numpy.zeros((8,4))
+        self.nn = sknn.mlp.BaseMLP(
+                    layers=[("LinearGaussian",)], learning_rate=0.001,
+                    n_iter=None, n_stable=1, f_stable=0.1,
+                    valid_size=0.25)
+
+        self.nn._fit(a_in, a_out)
+
 
 class TestTrainingOutput(unittest.TestCase):