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