Skip to content

A toy implementation of Neural Graph Collaborative filtering

Notifications You must be signed in to change notification settings

zyy598/NGCF-pytorch

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

NGCF

A pytorch toy implementation of Neural Graph Collaborative filtering

link:https://arxiv.org/pdf/1905.08108.pdf

Tested on dataset movielens 100k

Details

Add three transform layer to yield predictions of ratings

No node and message drop out

Evaluation

Train 0.8 test 0.2

SVD dim 50 RMSE 0.931

NCF dim 64 layers [128,64,32,8] RMSE 0.928

NGCF dim 64 layers [64,64,64] RMSE 0.896

About

A toy implementation of Neural Graph Collaborative filtering

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Languages

  • Python 89.2%
  • Perl 5.7%
  • Shell 5.1%