Skip to content

To predict if a customer will like a movie or not depending on his past ratings and preferences using RBMs.

Notifications You must be signed in to change notification settings

TEJASNARAYANS/Restricted-Boltzmann-Machines

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Restricted-Boltzmann-Machines

To predict if a customer will like a movie or not depending on his past ratings and preferences using RBMs. The given dataset includes movie dataset of over 4000 movies which have been rated by over 1000+ users.

Algorithm for the code

  1. Importing the dataset of movies,users and ratings
  2. Preparing the training and test set by converting them into an array.
  3. Obtaining the number of users and movies
  4. Converting the data ninto array with users in lines and movies in columns
  5. Converting the ratings into binary(0/1) Liked or Disliked

To run the code

  1. Set all the given files into the directory
  2. Run the rbm.py file

About

To predict if a customer will like a movie or not depending on his past ratings and preferences using RBMs.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published