dmlc--dgl
Minjie Wang
f13b9b6212
* WIP: api * dgl.sampling, dgl.data * dgl.sampling; dgl.dataloading * sampling packages * convert * subgraph * deprecate * subgraph APIs * All docstrings for convert/subgraph/transform * almost all funcs under dgl namespace * WIP: DGLGraph * done graph query * message passing functions * lint * fix merge error * fix test * lint * fix Co-authored-by: Quan Gan <coin2028@hotmail.com>
26 行
786 B
Python
26 行
786 B
Python
"""The ``dgl.dataloading`` package contains:
|
|
|
|
* Data loader classes for iterating over a set of nodes or edges in a graph and generates
|
|
computation dependency via neighborhood sampling methods.
|
|
|
|
* Various sampler classes that perform neighborhood sampling for multi-layer GNNs.
|
|
|
|
* Negative samplers for link prediction.
|
|
|
|
For a holistic explanation on how different components work together.
|
|
Read the user guide :ref:`guide-minibatch`.
|
|
|
|
.. note::
|
|
This package is experimental and the interfaces may be subject
|
|
to changes in future releases. It currently only has implementations in PyTorch.
|
|
"""
|
|
from .neighbor import *
|
|
from .dataloader import *
|
|
|
|
from . import negative_sampler
|
|
|
|
from .. import backend as F
|
|
|
|
if F.get_preferred_backend() == 'pytorch':
|
|
from .pytorch import *
|