search

snap-stanford / ogb

Benchmark datasets, data loaders, and evaluators for graph machine learning
https://ogb.stanford.edu
MIT License
1.89k stars 397 forks source link

Some question in the linkproppred/collab /gnn.py #443

Closed sktsherlock closed 12 months ago

sktsherlock commented 1 year ago

When loading the data: dataset = PygLinkPropPredDataset(name='ogbl-collab'); data = dataset[0]; edge_index = data.edge_index; I wonder the edge_index means what? Whether it represents the complete topology or the topology of the training set. If the latter, why is it not the same as split_edge['train']['edge']?

weihua916 commented 1 year ago

Topology of the training set. It is the same as split_edge['train']['edge'] except that data.edge_index is undirected.