migalkin
commented
5 months ago Triples to be predicted are stored in the PyG valid/test data object under target_edge_index and target_edge_type. For example, in transductive FB15k237 they are stored like this:
And for inductive Grail datasets like this:
During evaluation, we create a dataloader from them and loop through the triples
I didn't understand how to define the target triples that will be predicted. Could you enlighten me on this?