changeset 977:9cac1ecaeef7

mcRBM - changed init of U to match M'A.R's code
author James Bergstra <bergstrj@iro.umontreal.ca>
date Mon, 23 Aug 2010 16:04:10 -0400
parents 4cbd65cf902d
children ab4bc97ca060
files pylearn/algorithms/mcRBM.py
diffstat 1 files changed, 2 insertions(+), 1 deletions(-) [+]
line wrap: on
line diff
--- a/pylearn/algorithms/mcRBM.py	Mon Aug 23 16:03:38 2010 -0400
+++ b/pylearn/algorithms/mcRBM.py	Mon Aug 23 16:04:10 2010 -0400
@@ -265,7 +265,8 @@
         # initialization taken from Marc'Aurelio
 
         return cls(
-                U = numpy_project_onto_ball(rng.randn(n_I, n_F).T).T,
+                #U = numpy_project_onto_ball(rng.randn(n_I, n_F).T).T,
+                U = 0.2 * rng.randn(n_I, n_F),
                 W = rng.randn(n_I, n_J)/np.sqrt((n_I+n_J)/2),
                 a = np.ones(n_I)*(-2),
                 b = np.ones(n_K)*2,