Skip to content

Latest commit

 

History

History
10 lines (9 loc) · 655 Bytes

File metadata and controls

10 lines (9 loc) · 655 Bytes

Collaborative-filtering-on-MovieLens

This repository has implementation of following algorithms for collaborative filtering on movielens dataset

  1. Belief propogation based on Pairwise Markov Random Field
  2. Restricted Boltzman Machine
  3. Deep Belief Network
  4. Stacked Autoencoder. (An implementation of the paper Training Deep AutoEncoders for Collaborative Filtering).
  5. Variational Autoencoder

Dataset:

MovieLens dataaset is used for the task of collaborative filtering. This dataset has a set of users who have given ratings to a set of movies.