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

GSgnnNodeDataLoaderBase

The base dataloader class for node tasks.

GSgnnEdgeDataLoaderBase

The base dataloader class for edge tasks.

GSgnnLinkPredictionDataLoaderBase

The base class of link prediction dataloader.

DataSets

GSgnnNodeTrainData

Training data for node tasks

GSgnnNodeInferData

Inference data for node tasks

GSgnnEdgeTrainData

Edge prediction training data

GSgnnEdgeInferData

Edge prediction inference data

DataLoaders

GSgnnNodeDataLoader

Minibatch dataloader for node tasks

GSgnnEdgeDataLoader

The minibatch dataloader for edge prediction

GSgnnLinkPredictionDataLoader

Link prediction minibatch dataloader