SdA.py

Yusuke Sugomori · Yusuke Sugomori · commit 323b6e1c75a7 · 2013-02-13T12:42:13.000+09:00
diff --git a/SdA.py b/SdA.py
@@ -116,7 +116,6 @@ def predict(self, x):
         
         for i in xrange(self.n_layers):
             sigmoid_layer = self.sigmoid_layers[i]
-            # rbm_layer = self.rbm_layers[i]
             layer_input = sigmoid_layer.output(input=layer_input)
 
         out = self.log_layer.predict(layer_input)
@@ -153,7 +152,7 @@ def test_SdA(pretrain_lr=0.1, pretraining_epochs=1000, corruption_level=0.3, \
 
     # construct SdA
     sda = SdA(input=x, label=y, \
-              n_ins=20, hidden_layer_sizes=[10, 10], n_outs=2, numpy_rng=rng)
+              n_ins=20, hidden_layer_sizes=[15, 15], n_outs=2, numpy_rng=rng)
 
     # pre-training
     sda.pretrain(lr=pretrain_lr, corruption_level=corruption_level, epochs=pretraining_epochs)
