AshwinRJ / Restricted-Boltzmann-Machine

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Restricted-Boltzmann-Machine

This is an implementation of Restricted Boltzmann Machine (RBMs). The following links are useful:

http://image.diku.dk/igel/paper/AItRBM-proof.pdf - Intro to RBMs

https://link.springer.com/chapter/10.1007/978-3-642-35289-8_32 - Training RBMs

In the notebook, a abstract class called "Data Provider" provides a skeleton for any class that implements providing data batches to RBM. It is then used to feed MINST data to the RBM.

All the outputs at each training step are also shown.

About

License:MIT License


Languages

Language:Jupyter Notebook 100.0%