Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
talkingwallace committed May 23, 2019
1 parent 6944237 commit ccc2774
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,14 +1,20 @@
# GraphNN
A pytorch implementation of Neural Graph Collaborative filtering

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

Tested on toy dataset movielens 100k

# Details

Add three transform layer to yield prediction of ratings

# Evaluation
Train 0.9 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.914

0 comments on commit ccc2774

Please sign in to comment.