Use GNN to see how it labels a single instance of a graph

[emiserry (steven)]

Hi,

I’m completely new to pytorch and pytorch geometric and came across this colab tutorial

Google Colaboratory

My question is how I would go about giving the trained model a single MUTAG graph as an input to see how it classifies it. As an output I’d simply like to get the classification/label so I can compare it to the actual label.