RestrictedBoltzmannMachinesHDF5.jl

Author cossio
Popularity
0 Stars
Updated Last
4 Months Ago
Started In
June 2023

RestrictedBoltzmannMachinesHDF5 Julia package

Save and load Restricted Boltzmann machine objects as HDF5 files.

Installation

This package is registered. Install with:

import Pkg
Pkg.add("RestrictedBoltzmannMachinesHDF5")

Example

using RestrictedBoltzmannMachines: RBM, Binary # from https://github.com/cossio/RestrictedBoltzmannMachines.jl
using RestrictedBoltzmannMachinesHDF5: save_rbm, load_rbm # from this package
rbm = RBM(Binary(; θ=randn(2,3)), Binary(; θ=randn(4)), randn(2,3,4)) # create an RBM
path = save_rbm(tempname(), rbm) # save RBM to a temporary path
loaded_rbm = load_rbm(path) # load RBM from saved file

Related

RestrictedBoltzmannMachines.jl package, which defines the RBM object.

Used By Packages