← all repositories

yell/boltzmann-machines

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

852 stars Jupyter Notebook ML Frameworks
boltzmann-machines
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.