Mercurial > pylearn

changeset 1350:d957155264da

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
mcRBM - added norm_doctoring parameter
author James Bergstra <bergstrj@iro.umontreal.ca>
date Thu, 21 Oct 2010 23:12:38 -0400
parents 0d55f8f0aedc
children 6402b3309ece
files pylearn/algorithms/mcRBM.py
diffstat 1 files changed, 4 insertions(+), 2 deletions(-) [+]
line wrap: on
line diff
--- a/pylearn/algorithms/mcRBM.py	Thu Oct 21 14:47:49 2010 -0400
+++ b/pylearn/algorithms/mcRBM.py	Thu Oct 21 23:12:38 2010 -0400
@@ -618,7 +618,8 @@
             W_range=0.05,
             a_ival=0,
             b_ival=2,
-            c_ival=-2):
+            c_ival=-2,
+            norm_doctoring=(1.0/192, .5)):
         n_F, n_K = Pval.shape
         if not hasattr(rng, 'randn'):
             rng = np.random.RandomState(rng)
@@ -628,7 +629,8 @@
                 a = sharedX(np.ones(n_I)*a_ival,'a'),
                 b = sharedX(np.ones(n_K)*b_ival,'b'),
                 c = sharedX(np.ones(n_J)*c_ival,'c'),
-                P = sharedX(Pval, 'P'),)
+                P = sharedX(Pval, 'P'),
+                norm_doctoring=norm_doctoring)
         rval._params = [rval.U, rval.W, rval.a, rval.b, rval.c, rval.P]
         return rval