/examples/pytorch/dgi

Deep Graph Infomax (DGI)

• Paper link: https://arxiv.org/abs/1809.10341
• Author's code repo (in Pytorch): https://github.com/PetarV-/DGI

Dependencies

• PyTorch 0.4.1+
• requests

pip install torch requests

How to run

Run with following:

python3 train.py --dataset=cora --gpu=0 --self-loop

python3 train.py --dataset=citeseer --gpu=0

python3 train.py --dataset=pubmed --gpu=0

Results

• cora: ~81.6 (81.2-82.1) (paper: 82.3)
• citeseer: ~69.4 (paper: 71.8)
• pubmed: ~76.1 (paper: 76.8)