🆕 dgl.graphbolt¶

dgl.graphbolt is a dataloading framework for GNN that provides well-defined APIs for each stage of the data pipeline and multiple standard implementations.

APIs¶

`DataLoader`	Multiprocessing DataLoader.
`Dataset`	An abstract dataset which provides abstraction for accessing the data required for training.
`Task`	An abstract task which consists of meta information and Train/Validation/Test Set.
`ItemSet`	A wrapper of iterable data or tuple of iterable data.
`ItemSetDict`	Dictionary wrapper of ItemSet.
`ItemSampler`	A sampler to iterate over input items and create subsets.
`DistributedItemSampler`	A sampler to iterate over input items and create subsets distributedly.
`NegativeSampler`	A negative sampler used to generate negative samples and return a mix of positive and negative samples.
`SubgraphSampler`	A subgraph sampler used to sample a subgraph from a given set of nodes from a larger graph.
`SampledSubgraph`	An abstract class for sampled subgraph.
`MiniBatch`	A composite data class for data structure in the graphbolt.
`MiniBatchTransformer`	A mini-batch transformer used to manipulate mini-batch
`DGLMiniBatch`	A data class designed for the DGL library, encompassing all the necessary fields for computation using the DGL library.
`DGLMiniBatchConverter`	Convert a graphbolt mini-batch to a dgl mini-batch.
`Feature`	A wrapper of feature data for access.
`FeatureStore`	A store to manage multiple features for access.
`FeatureFetcher`	A feature fetcher used to fetch features for node/edge in graphbolt.
`CopyTo`	DataPipe that transfers each element yielded from the previous DataPipe to the given device.

Standard Implementations¶

`OnDiskDataset`	An on-disk dataset which reads graph topology, feature data and Train/Validation/Test set from disk.
`BuiltinDataset`	A utility class to download built-in dataset from AWS S3 and load it as `OnDiskDataset`.
`FusedCSCSamplingGraph`	A sampling graph in CSC format.
`UniformNegativeSampler`	Sample negative destination nodes for each source node based on a uniform distribution.
`NeighborSampler`	Sample neighbor edges from a graph and return a subgraph.
`LayerNeighborSampler`	Sample layer neighbor edges from a graph and return a subgraph.
`FusedSampledSubgraphImpl`	Sampled subgraph of FusedCSCSamplingGraph.
`BasicFeatureStore`	A basic feature store to manage multiple features for access.
`TorchBasedFeature`	A wrapper of pytorch based feature.
`TorchBasedFeatureStore`	A store to manage multiple pytorch based feature for access.
`GPUCachedFeature`	GPU cached feature wrapping a fallback feature.