RestrictedBoltzmannMachines.jl

Train and sample Restricted Boltzmann machines in Julia
Author cossio
Popularity
12 Stars
Updated Last
1 Year Ago
Started In
August 2020

RestrictedBoltzmannMachines Julia package

License codecov GitHub repo size GitHub code size in bytes

Train and sample Restricted Boltzmann machines in Julia. See the Documentation for details.

Installation

This package is registered. Install with:

import Pkg
Pkg.add("RestrictedBoltzmannMachines")

This package does not export any symbols. Since the name RestrictedBoltzmannMachines is long, it can be imported as:

import RestrictedBoltzmannMachines as RBMs

Related packages

Use RBMs on the GPU (CUDA):

Centered RBMs:

Adversarially constrained RBMs:

Citation

If you use this package in a publication, please cite:

  • Jorge Fernandez-de-Cossio-Diaz, Simona Cocco, and Remi Monasson. "Disentangling representations in Restricted Boltzmann Machines without adversaries." Physical Review X 13, 021003 (2023).

Or you can use the included CITATION.bib.