gigl.src.applied_tasks.test_tasks.academic#
Attributes#
Functions#
|
Cora graph is the graph in the first index in the returned dataset |
|
Module Contents#
- gigl.src.applied_tasks.test_tasks.academic.get_pyg_cora_dataset(store_at='/tmp/Cora')[source]#
Cora graph is the graph in the first index in the returned dataset i.e. the Planetoid object is subscriptable, data = dataset[0] Train and tests masks are defined by train_mask and test_mask` properties on data.
- Returns:
torch_geometric.datasets.planetoid.Planetoid
- Parameters:
store_at (str)
- Return type:
torch_geometric.datasets.planetoid.Planetoid