Fixed dbnexample. fixes rasmusbergpalm#5. Thanks.

DBN/dbnexamples.m

@@ -26,7 +26,7 @@
 dbn = dbnsetup(dbn, train_x, opts);
 dbn = dbntrain(dbn, train_x, opts);
 
-nn = dbnunfoldtonn(dbn);
+nn = dbnunfoldtonn(dbn, 10);
 
 nn.alpha  = 1;
 nn.lambda = 1e-4;
@@ -36,6 +36,5 @@
 nn = nntrain(nn, train_x, train_y, opts);
 [er, bad] = nntest(nn, test_x, test_y);
 
-%disp([num2str(er * 100) '% error']);
-printf('%5.2f% error', 100 * er)
+disp([num2str(er * 100) '% error']);
 figure; visualize(nn.W{1}', 1);

DBN/dbnunfoldtonn.m

@@ -1,9 +1,9 @@
-function nn = dbnunfoldtonn(dbn)
+function nn = dbnunfoldtonn(dbn, outputsize)
 %DBNUNFOLDTONN Unfolds a DBN to a NN
-%   Takes a DBN structure, traverses all layers and assigns upwards weights
-%   and biases to an equally sized NN structure which it returns
+%   dbnunfoldtonn(dbn, outputsize ) returns the unfolded dbn with a final
+%   layer of size outputsize added.
 
-    nn = nnsetup(dbn.sizes);
+    nn = nnsetup([dbn.sizes outputsize]);
     for i = 1 : numel(dbn.rbm)
         nn.W{i} = dbn.rbm{i}.W;
         nn.b{i} = dbn.rbm{i}.c;

DBN/rbmtrain.m

@@ -3,7 +3,7 @@
     m = size(x, 1);
     numbatches = m / opts.batchsize;
 
-    assert(rem(numbatches, 1) ~= 0, 'numbatches not integer');
+    assert(rem(numbatches, 1) == 0, 'numbatches not integer');
 
 
     for i = 1 : opts.numepochs

NN/nntrain.m

@@ -7,9 +7,7 @@
 
     numbatches = m / batchsize;
 
-    if rem(numbatches, 1) ~= 0
-        error('numbatches not integer');
-    end
+    assert(rem(numbatches, 1) == 0, 'numbatches not integer');
 
     nn.rL = [];
     n = 1;