Skip to content

Python3 implementation of the Unsupervised Deep Learning Algorithm, Restricted Boltzmann Machine.

License

Notifications You must be signed in to change notification settings

Karan-Malik/RestrictedBoltzmannMachine

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Restricted Boltzmann Machine

Python3 implementation of the unsupervised deep learning algorithm Restricted Boltzmann Machine using Pytorch.

Overview

The Restricted Boltzmann is a generative and stochastic artificial neural network that is used to learn probability distributions over a set of inputs. It can be used both as an Unsupervised or a Supervised algorithm, depending on the task. It comprises of visible nodes (inputs) and hidden nodes, and uses the contrastive divergence algorithm for training.

It is used as an unsupervised learning algorithm in this task and is implemented using Pytorch, an optimized tensor library for deep learning.

For theory and working of the Restricted Boltzmann Machine, check out this research paper by Asja Fischer and Christian Igel.

Dataset

The dataset used was taken from the Grouplens website, the Social Computing Research at the University of Minnesota.The data has also been uploaded in the repository under the names ml-1m and ml-100k.

To download the dataset click here

Copyright (c) 2020 Karan Malik