An open API service indexing awesome lists of open source software.

https://github.com/ttiagojm/dbn-tf2

Deep Belief Networks in Tensorflow 2
https://github.com/ttiagojm/dbn-tf2

convolutional-rbm crbm deep-belief-network generative-model rbm restricted-boltzmann-machine tensorflow2

Last synced: 3 months ago
JSON representation

Deep Belief Networks in Tensorflow 2

Awesome Lists containing this project

README

        

# DBN-TF2
Deep Belief Networks for image reconstruction in Tensorflow 2.


## Convolutional Restricted Boltzman Machine
`DBN/crbm.py` file contains the class where all logic and training code was implemented

The code for CRBM was adapted from [this](https://github.com/arthurmeyer/Convolutional_Deep_Belief_Network) repository. However, it's not being used the max pooling output, since we want to keep the spatial dimensions of the images.

## Discrete Restricted Boltzman Machine
`DBN/RBM/rbm.py` file contains the class where all logic and training code was implemented

Is not being used and the codes needs to be adapted for the new implementation of DBN