Code for train a Convolutional Restricted Boltzmann Machine. To test CRBM. run
python test.py
, assuming requirements are met,
Python 3.6+, and packages in requriements.txt
. install matplotlib>=2.0.0 for
running test.py
It is recommended for data being in the range of [-1,1]. RBM is a model for binary numbers.
For data far away from 0 (i.e. mean of data is greater than 3). Normalize the data before using CRBM, so that the data are centered on 0, with a variance 1.
Features(hidden) obtained from CRBM can be very noisy. consider denoising by using moving average window.