this commit was used to generate the training data for the full graph edge predictor and works with the corresponding commit on https://gitlab.laas.fr/roc/tim-luchterhand/nn-heuristic-implementation. Note that --learning and the --polarity-heuristic 1 need to be used