Skip to content

Using Neural Collaborative Filtering to recommend movies

Notifications You must be signed in to change notification settings

kxwtan/recommender-system

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Recommender System

Based on Neural Collaborative Filtering paper for recommending movie and Pinterest posts according to user interactions.

The model is built on PyTorch and combines the more simple Matrix Factorization and the more complex Multi-Layer perceptron to learn both linear and non-linear relationships between user interactions. It was made clear by the paper that this results in better HR@10 (Hit Rate @ 10) and NDCG@10 (Normalized Discounted Cumulative Gain @ 10) scores.

The model was able to achieve a HR@10 of almost 0.7 and a NDCG@10 of around 0.4 which aligns with the scores found in the research paper.

About

Using Neural Collaborative Filtering to recommend movies

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages