Dataloader module

Data loader module.

MARBLE.dataloader.loaders(data, par)

Loaders.

class MARBLE.dataloader.NeighborSampler(edge_index: Tensor | SparseTensor, sizes: List[int], node_idx: Tensor | None = None, num_nodes: int | None = None, return_e_id: bool = True, transform: Callable = None, **kwargs)

Neighbor Sampler.

sample(batch)

Sample.