topobench.dataloader.dataload_dataset module

Dataset class compatible with TBDataloader.

class DataloadDataset(data_lst)

Bases: Dataset

Custom dataset to return all the values added to the dataset object.

Parameters:

data_lstlist[torch_geometric.data.Data]

List of torch_geometric.data.Data objects.

__init__(data_lst)

get(idx)

Get data object from data list.

Parameters:

idxint

Index of the data object to get.

Returns:

tuple

Tuple containing a list of all the values for the data and the corresponding keys.

len()

Return the length of the dataset.

Returns:

int

Length of the dataset.