yell/boltzmann-machines
TensorFlow implementation of Restricted and Deep Boltzmann Machines with sklearn compatibility and examples on MNIST and CIFAR-10.

Velocity · 7d
+0.3
★ / day
Trend
→steady
star history
This repository provides generic and flexible implementations of Restricted Boltzmann Machines (RBM) and Deep Boltzmann Machines (DBM) using TensorFlow. It reproduces experiments from academic papers on deep learning and energy-based models, supporting variants like CD-k, PCD, and annealed importance sampling. The library offers a scikit-learn compatible API with extensive features for training and sampling.