-
Notifications
You must be signed in to change notification settings - Fork 28
Open
Description
Hi,thanks for sharing the code.
With your source code unmodified (dropout: 0.5, neg-weight: 0.5), I have tried on ml-1m and get the following results:
First col: Recall
Second col: NDCG
loss,loss_no_reg,loss_reg -20357.158033288044 -20357.158033288044 0.0
TopK: [10, 20, 50]
Recall@10: 0.1 NDCG@10: 0.04893161658667473
Recall@20: 0.16258278145695365 NDCG@20: 0.06466930639306495
Recall@50: 0.29817880794701984 NDCG@50: 0.09132394256584671
Which is much lower than in readme:
NDCG@5, 10, 20
0.2457 0.2475 0.2656
May I have your help to reproduce your results on ml1m. Thanks.
Metadata
Metadata
Assignees
Labels
No labels