tunsheng / Restricted-Boltzmann-Machines

Exploring pytorch & RBM model on Movielens dataset

Restricted Boltzmann Machines - Movielens

In this project, we try to build a movie recommendation system (binary) on movielens dataset using Restricted Boltzmann Machines. This project was done as a part of Deep Learning A-Z course on Udemy. In this project, I also explored using pytorch for developing the models.

The Jupyter notebook with RBM model can be found here.

The resources that helped me are:

• A Tutorial on Energy-Based Learning
• A fast learning algorithm for deep belief nets
• The wake-sleep algorithm for unsupervised neural networks
• Deep Boltzmann Machines