Skip to content

Latest commit

 

History

History
23 lines (14 loc) · 1.1 KB

README.md

File metadata and controls

23 lines (14 loc) · 1.1 KB
Raw

Graph Matching Networks for Learning the Similarity of Graph Structured Objects

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].

Running the code

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.