NIE-GCN: Neighbor Item Embedding-aware Graph Convolutional Network for Recommendation
python == 3.8.18
pytorch == 2.1.0 (cuda:12.1)
scipy == 1.10.1
numpy == 1.24.3
-
yelp2018: batch_size=1024, beta=1.0, l2=1e-4
-
amazon-book: batch_size=2048, beta=0.8, l2=1e-5