Skip to content

Deepest-Project/RBMAdvanced.jl

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

RBMAdvanced

Build Status

RBMAdvanced.jl is a library for training RBMs with methods starting from naive training to advanced training methods such as parallel tempering.

Usage Example

Below code trains a RBM on toy dataset and infers, using CD (contrastive divergence) method.

using RBMAdvanced

visibleSize = 2
hiddenSize = 3
dataset = [0 1; 0 0; 0 0]  # [0 1] for 1/3 probability, [0 0] otherwise

weight = randInitRBMWeight(visibleSize, hiddenSize)
cdConfig = TrainingConfig(
    usePersistentChain=false,
    betas=[1.0],
    k=1,
    epochs=100,
    learningRate=0.1
)
validateConfig(cdConfig)
trainRBM!(weight, dataset, cdConfig)
output = RBMAdvanced.inferRBM(weight, 1, 100, 100)

Check out our documentation to learn more about RBMs and how to train them.

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages