Skip to content

Latest commit

 

History

History
19 lines (10 loc) · 256 Bytes

File metadata and controls

19 lines (10 loc) · 256 Bytes

NGCF-simple

《Neural Graph Collaborative Filtering》simple implement.

Environments

  • python 3.7
  • pytorch-1.5

Example to run the codes

train model:

python main.py --dataset Yelp --hide_dim 16 --layer [16,16]