Data Transform

DataTransform class.

class topobenchmarkx.transforms.data_transform.DataTransform(transform_name, **kwargs)

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

Forward pass of the lifting.

Parameters:

datatorch_geometric.data.Data

The input data to be lifted.

Returns:

torch_geometric.data.Data

The lifted data.