Back in 2006, Netflix organized a competition with a grand prize of $1,000,000. The aim of the competition was to enhance the effectiveness of Netflix's Recommender system.
In this project, I will be leveraging deep learning to build a recommender system using the dataset from the competition. Graph Neural Networks have been shown to perform very well on tasks such as recommendation systems. I will use the LightGCN model (2020) for this mission.
- download the file movie_titles.csv.
- download the file combined_data_1.txt
- run the script "data_fix.py".
- delete all lines except the first 4500 in the file "movie_titles.csv".
Ensure you have the following dependencies installed:
- Python 3.9
- Torch 1.11.0
- Pandas 1.4.3
- NumPy 1.21.5
- PyG
- PyTorch Geometric - a library for graph neural networks (PyG).
- Creating a model that can learn with negative samples.
- Graph Neural Nets and Graph machine learning.
- Information about recommendation systems, such as collaborative filtering, etc.