graphstorm.dataloading
GraphStorm dataloading module includes a set of graph DataSets and DataLoaders for different graph machine learning tasks.
If users would like to customize DataLoaders, please extend those classes in the Base DataLoaders section and customize their abstract methods.
Base DataLoaders
The base dataloader class for node tasks. |
|
The base dataloader class for edge tasks. |
|
The base class of link prediction dataloader. |
DataSets
Training data for node tasks |
|
Inference data for node tasks |
|
Edge prediction training data |
|
Edge prediction inference data |
DataLoaders
Minibatch dataloader for node tasks |
|
The minibatch dataloader for edge prediction |
|
Link prediction minibatch dataloader |