This repository is the PyTorch implementation of "GLSearch: Maximum Common Subgraph Detection via Learning to Search" (ICML2021). For more details, please refer to our paper.
Package requirements are in requirements.txt. Please install required packages and versions.
Mark /model/OurMCS/ and /src/ as the source directories.
Get the datasets from: https://drive.google.com/drive/folders/1l7_WmO54gPHZ4YwzOqewZtZhWIxn44ZV?usp=sharing
Place the *.klepto files under a new directory /save/OurModelData/ (ex. /save/OurModelData/aids700nef_..._None.klepto).
Select 1 of 3 desired dataset on lines 62-77 of model/OurMCS/config.py by commenting out the rest.
Run the below command:
python3 main.py
Set training = True in config.py.
Run the below command:
python3 main.py