graphorge.gnn_base_model.data.graph_dataset.GNNGraphDatasetInMemory¶

class GNNGraphDatasetInMemory(dataset_samples)[source]¶

Bases: Dataset

Graph Neural Network graph data set (in-memory storage only).

_dataset_samples¶

Graph Neural Network graph data set samples data. Each sample is stored as a torch_geometric.data.Data object describing a homogeneous graph.

Type:: list

__len__(self):: Return size of data set (number of samples).

__getitem__(self, index)[source]¶: Return data set sample from corresponding index.

Constructor.

Parameters:: dataset_samples (list) – Graph Neural Network graph data set samples data. Each sample is stored as a torch_geometric.data.Data object describing a homogeneous graph.

List of Public Methods

Methods

__init__(dataset_samples)[source]¶

Constructor.

Parameters:: dataset_samples (list) – Graph Neural Network graph data set samples data. Each sample is stored as a torch_geometric.data.Data object describing a homogeneous graph.