Skip to content

Latest commit

 

History

History

graph_matching_networks

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 

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.