dgl.graphbolt.rst 1.25 KB
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
.. _apibackend:

🆕 dgl.graphbolt
=================================

.. currentmodule:: 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
-------------------------

.. autosummary::
    :toctree: ../../generated/
    :nosignatures:
    :template: classtemplate.rst

    Dataset
    Task
    ItemSet
    ItemSetDict
    ItemSampler
    NegativeSampler
    SubgraphSampler
25
26
    SampledSubgraph
    exclude_edges
27
    Feature
28
29
    Feature.read
    Feature.update
30
    FeatureStore
31
32
    FeatureStore.read
    FeatureStore.update
33
34
35
36
37
38
39
40
41
42
43
44
45
    FeatureFetcher
    CopyTo

DataLoaders
-----------

.. autosummary::
    :toctree: ../../generated/
    :nosignatures:
    :template: classtemplate.rst

    SingleProcessDataLoader
    MultiProcessDataLoader
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66

Standard Implementations
-------------------------

.. autosummary::
    :toctree: ../../generated/
    :nosignatures:
    :template: classtemplate.rst

    OnDiskDataset
    OnDiskTask
    OnDiskMetaData
    CSCSamplingGraph
    UniformNegativeSampler
    NeighborSampler
    LayerNeighborSampler
    SampledSubgraphImpl
    BasicFeatureStore
    TorchBasedFeature
    TorchBasedFeatureStore
    GPUCachedFeature