There are two folders named compare and train.

For training, open train folder and run command : python3 train.py

For convenient, we provide the trained model with name 'model_300.pth' in compare folder. 

For inference, open the compare folder and run command : python3 compare_benchmark.py

The resullts will be logged in a file results_benchmark_temp.txt in compare folder.

Note: The inference folder is synchronised with benchmark dataset in compare_benchmark.py, however, one can easily pass any networkx graph in the compare function.

