dmlc--dgl
Zihao Ye
9a0511c8e9
* upd * damn it * fuck * fuck pylint * fudge * remove some comments about MXNet * upd * upd * damn it * damn it * fuck * fuck * upd * upd * pylint bastard * upd * upd * upd * upd * upd * upd * upd * upd * upd
1013 B
1013 B
Graph Isomorphism Network (GIN)
- Paper link: arXiv OpenReview
- Author's code repo: https://github.com/weihua916/powerful-gnns.
Dependencies
- MXNet 1.5+
- sklearn
- tqdm
bash pip install torch sklearn tqdm
How to run
An experiment on the GIN in default settings can be run with
DGLBACKEND=mxnet python main.py
An experiment on the GIN in customized settings can be run with
DGLBACKEND=mxnet python main.py [--device 0 | --disable-cuda] --dataset COLLAB \
--graph_pooling_type max --neighbor_pooling_type sum
Results
Run with following with the double SUM pooling way: (tested dataset: "MUTAG"(default), "COLLAB", "IMDBBINARY", "IMDBMULTI")
DGLBACKEND=mxnet python main.py --dataset MUTAG --device 0 \
--graph_pooling_type sum --neighbor_pooling_type sum