Training on Graph edges in Pytorch Geometric

Is it possible to train a neural net against the edges of a graph in pytorch geometric?

From the readthedocs page of PyG:

data.y : Target to train against (may have an arbitrary shape), e.g. , node-level targets of shape.

But can data.y be the edge attributes of the graph (so data.edge_attr)?

1 Like