Skip to content
/ DGN Public

Implementation of Directional Graph Networks in PyTorch and DGL

Notifications You must be signed in to change notification settings

Saro00/DGN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

57 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Directional Graph Networks

Implementation of Directional Graph Networks in PyTorch and DGL.

method

Overview

We provide the implementation of the Directional Graph Networks (DGN) in PyTorch and DGL frameworks, along with scripts for running real-world benchmarks. The repository is organised as follows:

  • models contains:
    • pytorch contains the various GNN models implemented in PyTorch: the implementation of the aggregators, the scalers, the DGN layer and the directional aggregation matrix (eigen_agg).
    • dgl contains the DGN model implemented via the DGL library: aggregators, scalers, and DGN layer.
    • layers.py contains general NN layers used by the various models
  • realworld_benchmark contains various scripts from Benchmarking GNNs and Open Graph Benchmark to download the real-world benchmarks and train the DGN on them. In realworld_benchmark/README.md we provide instructions for runnning the experiments.

Reference

@article{beaini2020directional,
  title={Directional graph networks},
  author={Beaini, Dominique and Passaro, Saro and L{\'e}tourneau, Vincent and Hamilton, William L and Corso, Gabriele and Li{\`o}, Pietro},
  journal={arXiv preprint arXiv:2010.02863},
  year={2020}
}

License

MIT