rllm.dataloaderΒΆ

NeighborLoader

The random neighbor sampler, which allows for mini-batch training of GNNs on large-scale graphs where full-batch training is not feasible.

BRIDGELoader

BRIDGELoader is a specialized data loader for the BRIDGE model.

RelbenchLoader

DataLoader for RelBench dataset with heterogeneous neighbor sampling.