A pytorch toy implementation of Neural Graph Collaborative filtering
link:https://arxiv.org/pdf/1905.08108.pdf
Tested on dataset movielens 100k
Add three transform layer to yield predictions of ratings
No node and message drop out
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