Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
dataloading.rst 767 Bytes
Newer Older
Quan (Andy) Gan's avatar
[Feature] Edge DataLoader for edge classification & link prediction (#1828)  
Quan (Andy) Gan committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34 
.. _api-sampling:

dgl.dataloading
=================================

.. automodule:: dgl.dataloading

PyTorch node/edge DataLoaders
-----------------------------

.. autoclass:: pytorch.NodeDataLoader
.. autoclass:: pytorch.EdgeDataLoader

General collating functions
---------------------------

.. autoclass:: Collator
.. autoclass:: NodeCollator
.. autoclass:: EdgeCollator

Base Multi-layer Neighborhood Sampling Class
--------------------------------------------

.. autoclass:: BlockSampler

Uniform Node-wise Neighbor Sampling (GraphSAGE style)
-----------------------------------------------------

.. autoclass:: MultiLayerNeighborSampler

Negative Samplers for Link Prediction
-------------------------------------

.. autoclass:: negative_sampler.Uniform