This is the example code for the following ICML 2019 paper. If you use the code here please cite this paper.
Yujia Li, Chenjie Gu, Thomas Dullien, Oriol Vinyals, Pushmeet Kohli. Graph Matching Networks for Learning the Similarity of Graph Structured Objects. ICML 2019. [arXiv].
The code is in the format of a colab notebook, which includes:
- an example implementation of the model,
- an example graph similarity learning task,
- an example training loop, and
- some attention visualization tools.
To launch the notebook in Google colab, click here.
You can also download the notebook and run it locally with jupyter. The
notebook assumes you are on python 3 and have the latest (as of July 24,
2019) tensorflow, sonnet, numpy etc. installed. You can install the dependencies by running pip3 install --user -r requirements.txt.