Skip to content

Latest commit

 

History

History
 
 

dgi

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 

Deep Graph Infomax (DGI)

Dependencies

  • tensorflow 2.1+
  • requests
pip install tensorflow 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 (80.9-82.9) (paper: 82.3)
  • citeseer: ~70.2 (paper: 71.8)
  • pubmed: ~77.2 (paper: 76.8)