pub_soft/dgl

mirror of https://github.com/dmlc/dgl.git synced 2026-06-04 19:44:23 +08:00

Files

History

Andrei Ivanov 885b0eae92 Improving the DGI example. (#6034 )

2023-07-27 13:55:36 +08:00

..

dgi.py

[Misc] Rename number_of_edges and number_of_nodes to num_edges and num_nodes in examples. (#5492 )

2023-03-29 11:01:20 +08:00

gcn.py

[Misc] Black auto fix. (#4642 )

2022-09-26 21:47:11 +08:00

README.md

all demo use python-3 (#555 )

2019-05-23 10:38:05 +08:00

train.py

Improving the DGI example. (#6034 )

2023-07-27 13:55:36 +08:00

README.md

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)