graphorge.gnn_base_model.data.graph_dataset

Graph Neural Network graph data set.

Classes

GNNGraphDataset(torch.utils.data.Dataset)

Graph Neural Network graph data set.

GNNGraphDatasetInMemory(torch.utils.data.Dataset)

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

Functions

get_dataset_sample_files_from_dir

Get Graph Neural Network graph data set samples files from directory.

split_dataset

Randomly split data set into non-overlapping parts.

get_subset_indices_mapping

Get mapping from subset indexes to whole parent data set indices.

get_pyg_data_loader

Get Graph Neural Network graph data set PyG data loader.

write_graph_dataset_summary_file

Write summary data file for Graph Neural Network graph data set generation.

Functions

get_dataset_sample_files_from_dir(...)	Get Graph Neural Network graph data set samples files from directory.
get_pyg_data_loader(dataset[, batch_size, ...])	Get Graph Neural Network graph data set PyG data loader.
get_subset_indices_mapping(subset)	Get mapping from subset indexes to whole parent data set indices.
split_dataset(dataset, split_sizes[, ...])	Randomly split data set into non-overlapping subsets.
write_graph_dataset_summary_file(...[, ...])	Write summary data file for GNN-based data set generation.
write_summary_file(summary_directory[, ...])	Write summary data file with provided keyword-based parameters.

Classes

GNNGraphDataset(dataset_directory, ...[, ...])	Graph Neural Network graph data set.
GNNGraphDatasetInMemory(dataset_samples)	Graph Neural Network graph data set (in-memory storage only).