Train and sample Restricted Boltzmann machines in Julia.
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
Use RBMs on the GPU (CUDA):
Centered and standardized RBMs:
- https://github.com/cossio/CenteredRBMs.jl
- https://github.com/cossio/StandardizedRestrictedBoltzmannMachines.jl
Adversarially constrained RBMs:
Save RBMs to HDF5 files:
Stacked tempering:
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.