Skip to content

Restricted Boltzmann Machines (RBMs) in PyTorch

License

Notifications You must be signed in to change notification settings

jilinxia0825/pytorch-rbm

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Restricted Boltzmann Machines (RBMs) in PyTorch

Author: Gabriel Bianconi

Overview

This project implements Restricted Boltzmann Machines (RBMs) using PyTorch (see rbm.py). Our implementation includes momentum, weight decay, L2 regularization, and CD-k contrastive divergence. We also provide support for CPU and GPU (CUDA) calculations.

In addition, we provide an example file applying our model to the MNIST dataset (see mnist_dataset.py). The example trains an RBM, uses the trained model to extract features from the images, and finally uses a SciPy-based logistic regression for classification. It achieves 92.8% classification accuracy (this is obviously not a cutting-edge model).

About

Restricted Boltzmann Machines (RBMs) in PyTorch

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%