SS-00-SS/MBCGCN
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
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