Mercurial > pylearn

diff pylearn/algorithms/mcRBM.py @ 1273:7bb5dd98e671

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
mcRBM - added hack to main script to pickle final model and sampler
author James Bergstra <bergstrj@iro.umontreal.ca>
date Wed, 08 Sep 2010 13:05:02 -0400
parents ba25c6e4f55d
children f0129e37a8ef
line wrap: on
line diff
--- a/pylearn/algorithms/mcRBM.py	Sat Sep 04 19:32:27 2010 -0400
+++ b/pylearn/algorithms/mcRBM.py	Wed Sep 08 13:05:02 2010 -0400
@@ -561,4 +561,10 @@
 
 if __name__ == '__main__':
     import pylearn.algorithms.tests.test_mcRBM
-    pylearn.algorithms.tests.test_mcRBM.test_reproduce_ranzato_hinton_2010(as_unittest=True)
+    rbm,smplr = pylearn.algorithms.tests.test_mcRBM.test_reproduce_ranzato_hinton_2010(
+            as_unittest=False,
+            n_train_iters=10)
+    import cPickle
+    cPickle.dump(rbm, open('mcRBM.rbm.pkl', 'w'), -1)
+    cPickle.dump(smplr, open('mcRBM.smplr.pkl', 'w'), -1)
+