GNN for electrical distribution systems, HELP

Hello, I’m Diego G and I’m interested in the application of graph neural networks in power flow in electrical distribution systems, or in state estimation. I have to admit that I’m new to this topic of graph neural networks, but I’m very interested.
I need your help to create a correctly structured algorithm or if you can help me with material so I can read it and implement it in my work. I’ve implemented a code but it seems to me that it’s not structured correctly.

I appreciate your help and your time in advance.