关于GCU、沐曦GPGPU、MLU、0卡V100资源4月7日恢复上架的公告>>> 关于共建具身智能开源数据集的倡议>>> 关于云脑任务中统一路径访问方式的公告>>> 关于将启智集群GPU资源迁移至智算集群的公告>>>

History

Theheavens 6506d3b885 nightly update		1 year ago
..
infograph_eval.py	[fix] model name	1 year ago

infograph_trainer.py	Chebnet (#73)	1 year ago

infograph_util.py	[fix] model name	1 year ago

readme.md	nightly update	1 year ago

readme.md

InfoGraph
- How to run
- Results

InfoGraph

Paper link: https://arxiv.org/pdf/1908.01000
Author's code repo (in Pytorch):
https://github.com/fanyun-sun/InfoGraph

How to run

Run with following:

# use tensorflow bakcend
TL_BACKEND=tensorflow python infograph_trainer.py --dataset=MUTAG 
TL_BACKEND=tensorflow python infograph_trainer.py --dataset=IMDB-BINARY
TL_BACKEND=tensorflow python infograph_trainer.py --dataset=REDDIT-BINARY

# use pytorch backend
TL_BACKEND=torch python infograph_trainer.py --dataset=MUTAG

# use paddle backend
TL_BACKEND=paddle python infograph_trainer.py --dataset=MUTAG 
TL_BACKEND=paddle python infograph_trainer.py --dataset=IMDB-BINARY
TL_BACKEND=paddle python infograph_trainer.py --dataset=REDDIT-BINARY

Results

Dataset	MUTAG	IMDB-B	REDDIT-B
Author's Code	89.0	73.0	82.5
DGL	89.88	72.7	88.5
GammaGL(tf)	89.2 ± 0.25	72.0	82.4
GammaGL(th)	90.65 ± 0.2	--.-	--.-
GammaGL(pd)	89.2 ± 0.53	72.13 ± 0.12	84.8
GammaGL(ms)	--.-	--.-	--.-