• This repository has been archived on 03/Oct/2019
  • Stars
    star
    153
  • Rank 243,368 (Top 5 %)
  • Language
    Python
  • License
    MIT License
  • Created over 7 years ago
  • Updated over 6 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

Restricted Boltzmann Machines (RBMs) in PyTorch

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).