This is a PyTorch re-implementation of the following ICML 2019 paper. If you feel this project helpful to your research, please give a star.
Yujia Li, Chenjie Gu, Thomas Dullien, Oriol Vinyals, Pushmeet Kohli. Graph Matching Networks for Learning the Similarity of Graph Structured Objects. ICML 2019. [arXiv].
torch >= 1.2.0
networkx>=2.3
numpy>=1.16.4
six>=1.12
The code includes:
- an example implementation of the model,
- an example graph similarity learning task,
- an example training loop.
Please use ./run.sh to run.
Deepmind-research (using TensorFlow and Sonnet)