GeniePath-BS: GeniePath with Bandit Sampler

Paper link: http://arxiv.org/abs/2006.05806

Datasets

The ogbn-proteins dataset will be downloaded in directory ./dataset automatically.

Dependencies

• tensorflow = 1.13.1
• ogb 1.1.1

Prepare

python cython_sampler/setup.py build_ext -i

How to run

python train.py --dataset ogbn-proteins --learning_rate 1e-3 --epochs 200 --hidden1 64 --neighbor_limit 10 --batchsize 256

or

source main.sh

Performance

We train our models for 200 epochs and report the rocauc on the test dataset.

dataset	mean	std	#experiments
ogbn-proteins	0.78253	0.00352	10