cossio / RestrictedBoltzmannMachines.jl

Train and sample Restricted Boltzmann machines in Julia
MIT License
14 stars 3 forks source link
julia machine-learning rbm

RestrictedBoltzmannMachines Julia package

Train and sample Restricted Boltzmann machines in Julia.

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 and standardized RBMs:

Adversarially constrained RBMs:

Save RBMs to HDF5 files:

Stacked tempering:

Citation

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

Or you can use the included CITATION.bib.