/NIE-GCN

NIE-GCN: Neighbor Item Embedding-aware Graph Convolutional Network for Recommendation

Primary LanguagePython

NIE-GCN

NIE-GCN: Neighbor Item Embedding-aware Graph Convolutional Network for Recommendation

Environment

python == 3.8.18
pytorch == 2.1.0 (cuda:12.1)
scipy == 1.10.1
numpy == 1.24.3

Settings

  • yelp2018: batch_size=1024, beta=1.0, l2=1e-4

  • amazon-book: batch_size=2048, beta=0.8, l2=1e-5