Skip to content

Movie recommendation system implemented in C++ and trained/evaluated on Netflix Prize dataset.

Notifications You must be signed in to change notification settings

ztizzlegaming/movie-recommendation-system

Repository files navigation

Movie Recommendation System

Dr. Michael Lamar, Zeyang Huang, Jordan Turley

Overview

We developed and implemented a recommender system for movies. We developed a co-occurrence statistical learning algorithm trained and evaluated on the Netflix Prize dataset. The Netflix Prize was a competition hosted by Netflix from 2006 to 2009 allowing anyone to develop an algorithm for predicting user ratings for films. Our algorithm is based on the Co-Occurrence Data Embedding algorithm.

Our implementation takes a genetic or biological approach to solve this problem. We begin by creating a random vector for each user and each movie-rating (each movie has five possible ratings, so each movie gets five vectors). We put these vectors on a high dimensional unit sphere or torus. One can basically think of putting a random point on a ball, like a basketball or a soccer ball. We go through each user-movie-rating pair and attract the two closer together. We also repel these vectors away from random vectors to avoid all vectors converging to the same point. The user vector is repelled from a given number of random movie-rating vectors, and the movie-rating vector is repelled from a random number of user vectors. Then, the user vector is repelled from a random user vector, and the movie-rating vector is repelled from a random movie-rating vector.

A visual example of this is shown below:

Image of sphere and torus

Evaluation

The dataset is divided randomly into three sets: training, validation, and test. The training set is used for training, the validation set is used to tune our hyperparameters, and the test set will be used to find our final performance rating for the model.

The probability is calculated for each movie-rating and the expected value is calculated, which is our final prediction. The mean squared error is calculated to see how well our algorithm is doing. Currently, the best RMSE we have achieved is 0.943463.

Below, we see a real example of our algorithm learning a user's movie preferences. We selected a user that gave the movie "The Empire Strikes Back" a rating of 3. On the left is the predictions of the algorithm with no training, and on the right is the predictions after several iterations of training. Our algorithm begins predicting very close to the marginal distributions since the initialization is random, but after training, our algorithm learns that the user would likely give this movie a lower rating. The learning isn't perfect, but we do much better than we would do by predicting the marginal distribution. The video can be seen here or the start and finish can be seen below:

Image of learning

Poster

We presented a poster at the 2019 Joint Math Meetings in Baltimore, MD. The full resolution poster is here.

Image of poster

How to use

First, clone this repository using the following code:

git clone https://github.com/ztizzlegaming/netflix-prize

One must also have the Netflix Prize dataset, which can be found here. For ease, put this file in the same netflix-prize folder.

To compile the code, use the following command:

g++ main.cpp -o netflix -std=c++11 -O3

To run, use the following command:

./netflix input.txt netflix_data_c.bin 100480507

The first argument is the input file. The second is the dataset. The third is the number of data points to consider. 100,480,507 is the total number of data points. Fewer data points can be used for the algorithm to run more quickly.

When using the entire dataset, the program uses about 4.5 GB of RAM and takes about 20 minutes for each iteration.

Input File

The input file is simply a text file which is used for configuring several of the hyperparameters used in the model.

dimensions
η
φ (users)
φ (movie-ratings)
iterations
repulsion number

Dimensions is the number of dimensions that the user and movie-rating vectors are be. The model seems to perform better with more dimensions, so upwards of 40 or 50 works best. However, more dimensions does increase runtime.

η is the initial value of η. η is basically the power of the attraction and repulsion. A small η means the vectors will not move much when being attracted or repelled, but a large η means they will be moved significantly. η must be between zero and one. η = 0 means the vectors do not move at all, and η = 1 means when a vector is moved, it will move exactly to the spot of the vector it is being attracted to. We tend to keep η small: around 0.01.

φ is the rate that the vector movement slows down. Over time, the power of the attraction and repulsion decreases. Smaller values of φ cause the power to decrease very quickly, while larger values of φ cause the power to decrease more slowly. As there are 480,189 user vectors and 17,770 * 5 = 88,850 movie-rating vectors, we see that there are about 5.4 times as many users as movie-ratings. To keep the slow-down at the same rate, we tend to keep the movie-rating φ 5.4 times as large as the user φ.

Iterations is the number of passes we will take over the training data moving the vectors before terminating.

Repulsion number is the number of movie-rating vectors the user vector is repelled from and the number of user vectors the movie-rating vector is repelled from when doing attraction and repulsion. This does increase runtime, so we usually don't go above ten.

input.txt contains the current best hyperparameters we have found.

About

Movie recommendation system implemented in C++ and trained/evaluated on Netflix Prize dataset.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages