This directory contains the code for the baselines listed as Deep+MST
, Deep+ILP
and Deep+Greedy
in our paper.
The MST solver is implemented in C++ (mst.cpp
). Please build it first:
g++ mst.cpp -o mst --std=c++11
Following irit-stac, we also use SCIP as the ILP solver. Please download it and update the path argument scip_path
in main.py
.
python main.py --method=mst|ilp|greedy --train