矩池云上复现论文NeuralGraphCollaborativeFiltering环境复现

矩池云上复现论文 Neural Graph Collaborative Filtering 环境复现

Neural Graph Collaborative Filtering (NGCF) is a new recommendation framework based on graph neural network, explicitly encoding the collaborative signal in the form of high-order connectivities in user-item bipartite graph by performing embedding propagation.

https://github.com/xiangwang1223/neural_graph_collaborative_filtering


bash /public/script/switch_conda_source.sh


git clone https://github.com/xiangwang1223/neural_graph_collaborative_filtering.git
bash /public/script/switch_conda_source.sh
conda create -n py36 python&＃61;3.6.5


# To activate this environment, use
#
# $ conda activate py36
#
# To deactivate an active environment, use
#
# $ conda deactivate


conda activate py36


pip install tensorflow-gpu&＃61;&＃61;1.8.0
pip install scipy&＃61;&＃61;1.1.0
pip install scikit-learn&＃61;&＃61;0.19.1
pip install numpy&＃61;&＃61;1.14.3


cd /neural_graph_collaborative_filtering/NGCF/
python NGCF.py


Gowalla dataset

python NGCF.py --dataset gowalla --regs [1e-5] --embed_size 64 --layer_size [64,64,64] --lr 0.0001 --save_flag 1 --pretrain 0 --batch_size 1024 --epoch 400 --verbose 1 --node_dropout [0.1] --mess_dropout [0.1,0.1,0.1]


Amazon-book dataset

python NGCF.py --dataset amazon-book --regs [1e-5] --embed_size 64 --layer_size [64,64,64] --lr 0.0005 --save_flag 1 --pretrain 0 --batch_size 1024 --epoch 200 --verbose 50 --node_dropout [0.1] --mess_dropout [0.1,0.1,0.1]