DistEdgeDataLoaderο
- class dgl.dataloading.DistEdgeDataLoader(g, eids, graph_sampler, device=None, **kwargs)[source]ο
Bases:
DistDataLoader
Sampled graph data loader over edges for distributed graph storage.
It wraps an iterable over a set of edges, generating the list of message flow graphs (MFGs) as computation dependency of the said minibatch for edge classification, edge regression, and link prediction, on a distributed graph.
All the arguments have the same meaning as the single-machine counterpart
dgl.dataloading.DataLoader
except the first argumentg
which must be adgl.distributed.DistGraph
.- Parameters:
g (DistGraph) β The distributed graph.
eids β See
dgl.dataloading.DataLoader
.graph_sampler β See
dgl.dataloading.DataLoader
.device β See
dgl.dataloading.DataLoader
.kwargs β See
dgl.dataloading.DataLoader
.
See also