项目文件夹

文件

T

文件历史

xiang song(charlie.song) b347590a37 [Dataset] Citation graph (#1902 )

* citation graph

* GCN example use new citatoin dataset

* mxnet gat

* triger

* Fix

* Fix gat

* fix

* Fix tensorflow dgi

* Fix appnp, graphsage for mxnet

* fix monet and sgc for mxnet

* Fix tagcn

* update sgc, appnp

Co-authored-by: Ubuntu <ubuntu@ip-172-31-51-214.ec2.internal>

2020-08-01 19:25:50 +08:00

dgi.py

[Tensorflow] Several nn & example (#1191 )

2020-01-19 22:48:40 +08:00

gcn.py

[Tensorflow] Several nn & example (#1191 )

2020-01-19 22:48:40 +08:00

README.md

[Tensorflow] Several nn & example (#1191 )

2020-01-19 22:48:40 +08:00

train.py

[Dataset] Citation graph (#1902 )

2020-08-01 19:25:50 +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

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)