mirror of
https://github.com/dmlc/dgl.git
synced 2026-06-05 19:54:25 +08:00
168a88e570
* neighbor sampler data loader first commit * more commit * nodedataloader * fix * update RGCN example * update OGB * fixes * fix minibatch RGCN crashing with self loop * reverting gatconv test code * fix * change to new solution that doesn't require tf dataloader * fix * lint * fix * fixes * change doc * fix docstring * docstring fixes * return seeds and input nodes from data loader * fixes * fix test * fix windows build problem * add pytorch wrapper * fixes * add pytorch wrapper * add unit test * add -1 support to sample_neighbors & fix docstrings * docstring fix * lint * add minibatch rgcn evaluations Co-authored-by: xiang song(charlie.song) <classicxsong@gmail.com> Co-authored-by: Tong He <hetong007@gmail.com>
48 lines
1.1 KiB
ReStructuredText
48 lines
1.1 KiB
ReStructuredText
.. _api-sampling:
|
|
|
|
dgl.sampling
|
|
=================================
|
|
|
|
.. automodule:: dgl.sampling
|
|
|
|
Sampling algorithms on graphs.
|
|
|
|
Random walk sampling functions
|
|
------------------------------
|
|
|
|
.. autosummary::
|
|
:toctree: ../../generated/
|
|
|
|
random_walk
|
|
pack_traces
|
|
|
|
Neighbor sampling functions
|
|
---------------------------
|
|
|
|
.. autosummary::
|
|
:toctree: ../../generated/
|
|
|
|
sample_neighbors
|
|
select_topk
|
|
|
|
PyTorch DataLoaders with neighborhood sampling
|
|
----------------------------------------------
|
|
.. autoclass:: pytorch.NeighborSamplerNodeDataLoader
|
|
|
|
Builtin sampler classes for more complicated sampling algorithms
|
|
----------------------------------------------------------------
|
|
.. autoclass:: RandomWalkNeighborSampler
|
|
.. autoclass:: PinSAGESampler
|
|
|
|
Neighborhood samplers for multilayer GNNs
|
|
-----------------------------------------
|
|
.. autoclass:: MultiLayerNeighborSampler
|
|
|
|
Data loaders for minibatch iteration
|
|
------------------------------------
|
|
.. autoclass:: NodeCollator
|
|
|
|
Abstract class for neighborhood sampler
|
|
---------------------------------------
|
|
.. autoclass:: BlockSampler
|