Skip to content

HarshdeepGupta/recommender_pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Neural Collaborative Filtering in PyTorch

This repository shows you how to implement a Recommender System in PyTorch from scratch.

Prepare the dataset

The file u.data is the one picked from MovieLens 100K archive. Run the following command to get the train and test split for the dataset.

python train_test_split.py

Run the Item Popularity Model

The following command runs the model and prints the metrics

python itempop.py

Run the deep learning based Model

The following command runs the model and prints the metrics

python MLP.py

About

A recommender system in PyTorch from scratch.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages