dgl.dataloading.NodeDataLoader

class dgl.dataloading.NodeDataLoader(graph, indices, graph_sampler, device=None, use_ddp=False, ddp_seed=0, batch_size=1, drop_last=False, shuffle=False, use_prefetch_thread=None, use_alternate_streams=None, pin_prefetcher=None, use_uva=False, use_cpu_worker_affinity=False, cpu_worker_affinity_cores=None, **kwargs)[source]

(DEPRECATED) Sampled graph data loader over a set of nodes.

Deprecated since version 0.8: The class is deprecated since v0.8, replaced by DataLoader.

__init__(graph, indices, graph_sampler, device=None, use_ddp=False, ddp_seed=0, batch_size=1, drop_last=False, shuffle=False, use_prefetch_thread=None, use_alternate_streams=None, pin_prefetcher=None, use_uva=False, use_cpu_worker_affinity=False, cpu_worker_affinity_cores=None, **kwargs)

Initialize self. See help(type(self)) for accurate signature.

Methods

__init__(graph, indices, graph_sampler[, …])

Initialize self.

attach_data(name, data)

Add a data other than node and edge features for prefetching.

check_worker_number_rationality()

Attributes

multiprocessing_context