Similarly to the first GNN you implemented on MUTAG, learn a GNN model on the train dataset, predict on the test set and submit your results !