topobench.transforms.data_transform module

DataTransform class.

class DataTransform(transform_name, **kwargs)

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”.

__init__(transform_name, **kwargs)

forward(data)

Forward pass of the lifting.

Parameters:

datatorch_geometric.data.Data

The input data to be lifted.

Returns:

torch_geometric.data.Data

The lifted data.