partly fixed rnnrbm, but we will need to do some magic with the midi module to make it compatible with python 3

gyom · gyom · commit 53f246dc2cd7 · 2016-02-08T14:30:51.000-05:00
diff --git a/code/rnnrbm.py b/code/rnnrbm.py
@@ -3,6 +3,9 @@
 # RNN-RBM deep learning tutorial
 # More information at http://deeplearning.net/tutorial/rnnrbm.html
 
+from __future__ import print_function
+from six.moves import xrange
+
 import glob
 import os
 import sys
@@ -11,10 +14,8 @@
 try:
     import pylab
 except ImportError:
-    print (
-        "pylab isn't available. If you use its functionality, it will crash."
-    )
-    print "It can be installed with 'pip install -q Pillow'"
+    print ("pylab isn't available. If you use its functionality, it will crash.")
+    print("It can be installed with 'pip install -q Pillow'")
 
 from midi.utils import midiread, midiwrite
 import theano
@@ -257,12 +258,12 @@ def train(self, files, batch_size=100, num_epochs=200):
                         cost = self.train_function(sequence[i:i + batch_size])
                         costs.append(cost)
 
-                print 'Epoch %i/%i' % (epoch + 1, num_epochs),
-                print numpy.mean(costs)
+                print('Epoch %i/%i' % (epoch + 1, num_epochs))
+                print(numpy.mean(costs))
                 sys.stdout.flush()
 
         except KeyboardInterrupt:
-            print 'Interrupted by user.'
+            print('Interrupted by user.')
 
     def generate(self, filename, show=True):
         '''Generate a sample sequence, plot the resulting piano-roll and save
