Skip to content

jasonwbw/EffictiveRBM

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

67 Commits
 
 
 
 
 
 
 
 

Repository files navigation

EffictiveRBM

EffictiveRBM is a python version restricted boltzmann machine tools for solve problems or build more complex model. The effictive means this is a fast rbm tools, it accelerate by cython and multi-process. It has been used to build the autoencoder model which descript by Hinton.

How to Use

First, initialize an RBM with the desired number of visible, hidden units and process number.
Then train the rbm model with Iterator and params which descript in rbm.py

rbm = ParallelRBM(self.converter.dimensionality, 1000, 5)
rbm.train(DefaultBatchIterator(100, self.converter.train_images), max_epochs = 20, batch = 100)

Dependencies

  • NumPy: normal computing
  • Scipy: faster sigmoid computing

Optional dependencies

  • Cython: Only necessary to run the faster version. Version 0.17.1 or higher.
  • matplotlib: For plotting some data

Reference

License

The MIT License (MIT)

About

python version restricted boltzmann machine tools for solve problems or build more complex model

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages