/stacked-RBM

RBM module using pytorch. Assignment 2, part 4&5, CS6910: Deep Learning, IIT Madras

Primary LanguagePython

Restricted Bolzmann Machines

  • refer RBM.py for class implementation.
  • refer binary-binary.py for gaussian-binary.py for training code for real datasets.

Done in collaboration with github.com/DevSheth, running and testing one on windows machine.