Coding review for edge classification

Step 1 : install py torch geometric library

Step 2 : import the required modules for data processing
Step 3 : code the GNN class
Step 4 :
Using the table, create data object as tensor and define the labels
here
0-true friendship
1-uncertain friendship
2-false friendship
3-liberal friendship
Step 5 : train the model using training loop
Step 6 : predict the remaining labels using Trained Model