dont sample in the last hidden layer activation in RBM, fixes rasmusb…

…ergpalm#65
chvillap · Nov 12, 2013 · bb431c2 · bb431c2
1 parent a6499c3
commit bb431c2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/DBN/rbmtrain.m b/DBN/rbmtrain.m
@@ -14,7 +14,7 @@
             v1 = batch;
             h1 = sigmrnd(repmat(rbm.c', opts.batchsize, 1) + v1 * rbm.W');
             v2 = sigmrnd(repmat(rbm.b', opts.batchsize, 1) + h1 * rbm.W);
-            h2 = sigmrnd(repmat(rbm.c', opts.batchsize, 1) + v2 * rbm.W');
+            h2 = sigm(repmat(rbm.c', opts.batchsize, 1) + v2 * rbm.W');
 
             c1 = h1' * v1;
             c2 = h2' * v2;