forked from SS-00-SS/MBCGCN
-
Notifications
You must be signed in to change notification settings - Fork 0
toufuheyu/MBCGCN
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
######################## environment: Python 3.6 TensorFlow 1.14. ######################### Dataset:tmall Pre-training on GCN: python GCN.py --data_path Data/ --dataset 'view' --pretrain 0 --epoch 1000 --embed_size 64 --layer_size [64,64,64] --batch_size 2048 --regs [1e-4] --lr 0.001 --model-type 'gcn' --alg_type 'gcn' --adj_type 'pre' --save_flag 1 And then training at the RCGCN: python MBCGCN.py --data_path Data/ --dataset 'buy' --pretrain 1 --epoch 1000 --embed_size 64 --layer_size [64,64,64] --layer_size2 [64,64,64] --layer_size3 [64,64,64,64] --layer_size4 [64,64] --batch_size 2048 --regs [1e-4] --lr 0.001 --model-type 'mbcgcn' --alg_type 'mbcgcn' --adj_type 'pre' --save_flag 1
About
MBCGCN
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published
Languages
- Python 88.4%
- C++ 8.5%
- Cython 3.1%