GraphBatchDataset

class caldera.dataset.GraphBatchDataset(datalist, transform=None)[source]

Dataset that loads a caldera.data.GraphData list into.

caldera.data.GraphBatch instances, and then performs transform (if provided).

Parameters

  • datalist (List[GraphData]) –

  • transform (Union[Callable, TransformBase, None]) –

Creates a graph dataset.

Parameters

  • datalist (List[GraphData]) –

  • transform (Union[Callable, TransformBase, None]) –

__init__(datalist, transform=None)

Creates a graph dataset.

Parameters

  • datalist (List[GraphData]) –

  • transform (Union[Callable, TransformBase, None]) –

Methods

__init__(datalist[, transform])

Creates a graph dataset.

get(idx[, transform])

param transform

mem_sizes()

__init__(datalist, transform=None)

Creates a graph dataset.

Parameters

  • datalist (List[GraphData]) –

  • transform (Union[Callable, TransformBase, None]) –