Add the new rnnrbm tutorial in the normal test

Author: nouiz

code/rnnrbm.py

@@ -258,9 +258,13 @@ def generate(self, filename, show=True):
             pylab.title('generated piano-roll')
 
 
-if __name__ == '__main__':
+def test_rnnrbm(batch_size=100, num_epochs=200):
     model = RnnRbm()
-    model.train(glob.glob('../data/Nottingham/train/*.mid'))
+    model.train(glob.glob('../data/Nottingham/train/*.mid'),
+                batch_size=batch_size, num_epochs=num_epochs)
+
+if __name__ == '__main__':
+    test_rnnrbm()
     model.generate('sample1.mid')
     model.generate('sample2.mid')
     pylab.show()

code/test.py

@@ -10,6 +10,7 @@
 import logistic_sgd
 import mlp
 import rbm
+import rnnrbm
 import SdA
 
 
@@ -49,6 +50,8 @@ def test_rbm():
     rbm.test_rbm(training_epochs=1, batch_size=300, n_chains=1, n_samples=1,
                 n_hidden=20, output_folder='tmp_rbm_plots')
 
+def test_rnnrbm():
+    rnnrbm.test_rnnrbm(num_epochs=1)
 
 def speed():
     """