Skip to content

smrfeld/rbm-tensorflow-2

main
Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
rbm
 
 
 
 
 
 
 
 
 
 
 
 
 
 

RBMs revisited in TensorFlow 2

This repo. contains an implementation of RBMs in TensorFlow 2, (specifically: 2.3).

Dreaming digits:

  • Top row = original training data.
  • Bottom row = dreamed digits after CD sampling.

drawing

Histogram of weights and biases: drawing

Usage

python rbm_mnist.py

You will be prompted for more arguments.

Tests

Tests for various components are:

python test_gibbs.py
python test_loss.py
python test_rbm.py

Source

The implementation of the RBM (and the Gibbs sampler) is in the rbm folder.

About

RBMs revisited in TensorFlow 2

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages