Objects

Base classes and core data structures.

class unravel.utils.GraphDataset[source]

Bases:

Factory function that creates the appropriate dataset based on format.

Parameters:

format – Format specification (‘spektral’ or ‘pyg’). Defaults to ‘spektral’.
**kwargs – Arguments passed to the dataset constructor

Returns:

SpektralGraphDataset or PyGGraphDataset depending on format

Examples

# Spektral format (default) dataset = GraphDataset(graphs=spektral_graph_list, format=’spektral’)

# PyG format dataset = GraphDataset(graphs=pyg_data_list, format=’pyg’)

# From pickle files dataset = GraphDataset(pickle_file=’graphs.pickle.gz’, format=’pyg’)

from unravel.utils import GraphDataset

# Create dataset
dataset = GraphDataset(graphs=graphs, format="pyg")

# Split data
train, test, val = dataset.split_test_train_validation(4, 1, 1)