you can refer to https://tianchi.aliyun.com/dataset/162145 on Alibaba Tianchi platform to get the share dataset, and we also provide some other side information for future research. Then to run this code, please follow the commmand:
python train.py --batch_size 4096 --num_neighbors 2