-
Notifications
You must be signed in to change notification settings - Fork 0
wangxiaoyunNV/graphsaint
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
Notes: 1. "modules", "sampler", "train_sampling", and "utils.py" are implemented by DGL. https://github.com/dmlc/dgl/tree/master/examples/pytorch/graphsaint 2. "my_sampler1" and "my_train_sampling1" are implemented to integrate cuGraph with DGL: using cuGraph RW to sample subgraphs and train GCN model using DGL. 3. "my_sampler2" and "my_train_sampling2" are used to run deeper RW and use walk paths as subgraphs instead of using subgraphs induced from sampled nodes by RW. To run the code: 1. python train_sampling.py --gpu 0 --dataset ppi --sampler rw --num-roots 3000 --length 2 --num-repeat 50 --n-epochs 1000 --n-hidden 512 --arch 1-0-1-0 --dropout 0.1 2. python my_train_sampling1.py --gpu 0 --dataset ppi --sampler cugraph_rw --num-roots 3000 --length 2 --num-repeat 50 --n-epochs 1000 --n-hidden 512 --arch 1-0-1-0 --dropout 0.1 3. python my_train_sampling2.py --gpu 0 --dataset ppi --sampler cugraph_rw --num-roots 1000 --length 20 --num-repeat 50 --n-epochs 1000 --n-hidden 512 --arch 1-0-1-0 --dropout 0.1 4. python my_train_sampling3.py --gpu 0 --dataset ppi --sampler cugraph_ego --num-roots 3000 --length 2 --num-repeat 50 --n-epochs 1000 --n-hidden 512 --arch 1-0-1-0 --dropout 0.1
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published