GNN. For the graph neural network baseline, we followed the design of Allamanis et al. [1]} as close as possible. We reimplemented the GNN based on the reference implementation provided by the authors. The GNN consists of 8 message propagation layers with a skip connection between the first and the fourth layer and between the fourth and the eights layer. The node hidden size is set to 256.
