graphorge.gnn_base_model.data.graph_dataset.get_pyg_data_loader¶
- get_pyg_data_loader(dataset, batch_size=1, is_shuffle=False, is_verbose=False, **kwargs)[source]¶
Get Graph Neural Network graph data set PyG data loader.
- Parameters:
dataset ({GNNGraphDataset, list[str]}) – Graph Neural Network graph data set. Each sample corresponds to a torch_geometric.data.Data object describing a homogeneous graph. Accepts GNNGraphDataset or a list of data set samples files paths, where each sample file contains a torch_geometric.data.Data object describing a homogeneous graph.
batch_size (int, default=1) – Number of samples loaded per batch.
is_shuffle (bool, default=False) – Reshuffle data set at every epoch.
is_verbose (bool, default=False) – If True, enable verbose output.
**kwargs – Arguments of torch.utils.data.DataLoader.
- Returns:
data_loader – Graph Neural Network graph data set PyG data loader.
- Return type: