Skip to content
This repository was archived by the owner on Feb 21, 2025. It is now read-only.

Files

Latest commit

e452179 · Oct 13, 2022

History

History
This branch is 1690 commits behind dmlc/dgl:master.

dgi

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
May 23, 2019
Sep 26, 2022
Sep 26, 2022
Oct 13, 2022

Deep Graph Infomax (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)