topobench.transforms package#

Subpackages#

Submodules#

topobench.transforms.data_transform module#

DataTransform class.

class topobench.transforms.data_transform.DataTransform(transform_name, **kwargs)[source]#

Bases: BaseTransform

Abstract class to define a custom data lifting.

Parameters:
transform_namestr

The name of the transform to be used.

**kwargsdict

Additional arguments for the class. Should contain “transform_name”.

forward(data: Data) Data[source]#

Forward pass of the lifting.

Parameters:
datatorch_geometric.data.Data

The input data to be lifted.

Returns:
torch_geometric.data.Data

The lifted data.

Module contents#

This module contains the transforms for the topobench package.