“How’s that movie?” — Neural collaborative filtering with FastAI

Build a state-of-the-art movie recommendation system with just 10 lines of code

Source: Unsplash and 3Blue1Brown

The MovieLens 100K Dataset

System Setup

pip install jovian --upgrade     # Install the jovian library 
jovian clone a1b40b04f5174a18bd05b17e3dffb0f0 # Download notebook
cd movielens-fastai # Enter the created directory
jovian install # Install the dependencies
conda activate movielens-fastai # Activate virtual environment
jupyter notebook # Start Jupyter

Preparing the data

Neural collaborative filtering model

Source: FastAI Lesson 4
Source: Wikipedia

Training the model

Looking at some predictions

Source: Netflix

Save and commit

Further Reading

Founder, Jovian