The model training part is re-organized only till the end of "Graph Neural Networks" section. The rest is still work in progress. \

The first part of work is adapted from this GNN tutorial: https://uvadlc-notebooks.readthedocs.io/en/latest/tutorial_notebooks/tutorial7/GNN_overview.html