Fix for corrputor AI exceptions, tests that actually run fit().

aigamedev · May 17, 2015 · e365d94 · e365d94
1 parent 37717b7
commit e365d94
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/sknn/ae.py b/sknn/ae.py
@@ -25,7 +25,7 @@ def __init__(self,
                  units=None,
                  cost='msre',
                  tied_weights=False,
-                 corruption_level=None):
+                 corruption_level=0.5):
 
         assert warning is None, \
             "Specify layer parameters as keyword arguments, not positional arguments."

diff --git a/sknn/tests/test_ae.py b/sknn/tests/test_ae.py
@@ -21,12 +21,16 @@ def test_FitData(self):
 class TestParameters(unittest.TestCase):
 
     def test_CostFunctions(self):
+        X = numpy.zeros((8,4))
         for t in ['msre', 'mbce']:
-            AE(layers=[L("Sigmoid", units=8, cost=t)])
+            ae = AE(layers=[L("Sigmoid", units=8, cost=t)], n_iter=1)
+            ae.fit(X)
 
     def test_LayerTypes(self):
+        X = numpy.zeros((8,4))
         for l in ['autoencoder', 'denoising']:
-            AE(layers=[L("Sigmoid", type=l, units=8)])
+            ae = AE(layers=[L("Sigmoid", type=l, units=8)])
+            ae.fit(X)
 
     def test_UnknownCostFunction(self):
         assert_raises(NotImplementedError, L, "Sigmoid", cost="unknown")