data.dataloader
¶
Module Contents¶
Classes¶
Data loader which merges data objects from a |
|
Data loader which merges data objects from a |
|
Data loader which merges data objects from a |
-
class
data.dataloader.
DataLoader
(dataset, batch_size=1, shuffle=True, **kwargs)[source]¶ Bases:
torch.utils.data.DataLoader
Data loader which merges data objects from a
cogdl.data.dataset
to a mini-batch.
-
class
data.dataloader.
DataListLoader
(dataset, batch_size=1, shuffle=True, **kwargs)[source]¶ Bases:
torch.utils.data.DataLoader
Data loader which merges data objects from a
cogdl.data.dataset
to a python list.Note
This data loader should be used for multi-gpu support via
cogdl.nn.DataParallel
.
-
class
data.dataloader.
DenseDataLoader
(dataset, batch_size=1, shuffle=True, **kwargs)[source]¶ Bases:
torch.utils.data.DataLoader
Data loader which merges data objects from a
cogdl.data.dataset
to a mini-batch.Note
To make use of this data loader, all graphs in the dataset needs to have the same shape for each its attributes. Therefore, this data loader should only be used when working with dense adjacency matrices.