Skip to content

Latest commit

 

History

History
50 lines (31 loc) · 1.26 KB

DBM_MNIST.rst

File metadata and controls

50 lines (31 loc) · 1.26 KB

Deep Boltzmann machines on MNIST

Example for training a centered Deep Boltzmann machine on the MNIST handwritten digit dataset.

It allows to reproduce the results from the publication How to Center Deep Boltzmann Machines. Melchior et al. JMLR 2016..

Results

The code given below produces the following output that is quite similar to the results produced by an RBM.

The learned filters of the first layer

DBM filters of the first layer on MNIST

The learned filters of the second layer, linearly back projected

DBM filters of the second layer on MNIST

Some generated samples

AE filter on MNIST with contrastive penalty

See also RBM_MNIST_big.

Source code

images/download_icon.png
.. literalinclude:: ../../examples/DBM_MNIST.py