the difference between edge_index and edge_label_index #5189
-
Hi i am trying to implement a link prediction model and using https://github.com/pyg-team/pytorch_geometric/blob/master/examples/link_pred.py but i am unable to understand the difference between edge_index and edge_label index |
Beta Was this translation helpful? Give feedback.
Replies: 1 comment 3 replies
-
Hi, |
Beta Was this translation helpful? Give feedback.
Hi,
edge_index
is the default name for the message passing edges index, that is, the set of edges we allow the model to send messages across.edge_label_index
refers to the so called supervision edges: the ground truth values we want to compare our model's performance against, with consecutive loss calculation and backpropagation. You can read more here.