项目文件夹

文件

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

gat.py

[NN] nn modules & examples update (#890 )

2019-11-03 17:29:42 -05:00

README.md

[Hotfix] Fix GAT example and clarify the usage of early stop (#1065 )

2019-11-30 23:43:15 +08:00

train.py

[Dataset] Citation graph (#1902 )

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

utils.py

[Feature] Edge softmax on a subset of edges in the graph. (#842 )

2019-09-09 13:44:56 +08:00

README.md

Graph Attention Networks (GAT)

Paper link: https://arxiv.org/abs/1710.10903
Author's code repo: https://github.com/PetarV-/GAT.

Note that the original code is implemented with Tensorflow for the paper.

Dependencies

MXNet nightly build
requests

pip install mxnet --pre
pip install requests

Usage (make sure that DGLBACKEND is changed into mxnet)

DGLBACKEND=mxnet python3 train.py --dataset cora --gpu 0
DGLBACKEND=mxnet python3 train.py --dataset citeseer --gpu 0 --early-stop
DGLBACKEND=mxnet python3 train.py --dataset pubmed --gpu 0 --early-stop