As I’ve been looking around, I’ve come to the realisation that this is a supervised binary graph classification problem – so a graph convolutional network using the stellargraph library 3 should do the trick. Let me know if you think I’m totally wrong about this!