mudittiwari255 / Restricted-Boltzman-Machine-Simple-Implementation

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Restricted-Boltzman-Machine-Simple-Implementation

Objective is to map images of MNIST dataset (each image comprises of 784 pixels) to a lower dimension by reducting the Reconstruction error. Also, we want to demonstrate the reconstruction of each image. All the codes written are based on this great paper.

Usage

This Repo contains three files:

  1. boltzman.py : Contains utility function for one RBM.
  2. encoder.py : Contains utility functions for stacking RBMs.
  3. mnist.py : Contains the driver code to minimize reconstruction error and hence encoding the images.

Just keep every file in same path and run mnist.py.

(Dependencies : Tensor-flow, numpy).

About

License:MIT License


Languages

Language:Python 100.0%