topobench.transforms.feature_liftings package#

Feature lifting transforms with automated exports.

class Concatenation(**kwargs)#

Bases: BaseTransform

Lift r-cell features to r+1-cells by concatenation.

Parameters:
**kwargsoptional

Additional arguments for the class.

__init__(**kwargs)#
forward(data)#

Apply the lifting to the input data.

Parameters:
datatorch_geometric.data.Data | dict

The input data to be lifted.

Returns:
torch_geometric.data.Data | dict

The lifted data.

lift_features(data)#

Concatenate r-cell features to obtain r+1-cell features.

Parameters:
datatorch_geometric.data.Data | dict

The input data to be lifted.

Returns:
torch_geometric.data.Data | dict

The lifted data.

class Identity(**kwargs)#

Bases: BaseTransform

An identity transform that does nothing to the input data.

Parameters:
**kwargsoptional

Parameters for the base transform.

__init__(**kwargs)#
forward(data)#

Apply the transform to the input data.

Parameters:
datatorch_geometric.data.Data

The input data.

Returns:
torch_geometric.data.Data

The same data.

class ProjectionSum(**kwargs)#

Bases: BaseTransform

Lift r-cell features to r+1-cells by projection.

Parameters:
**kwargsoptional

Additional arguments for the class.

__init__(**kwargs)#
forward(data)#

Apply the lifting to the input data.

Parameters:
datatorch_geometric.data.Data | dict

The input data to be lifted.

Returns:
torch_geometric.data.Data | dict

The lifted data.

lift_features(data)#

Project r-cell features of a graph to r+1-cell structures.

Parameters:
datatorch_geometric.data.Data | dict

The input data to be lifted.

Returns:
torch_geometric.data.Data | dict

The data with the lifted features.

class Set(**kwargs)#

Bases: BaseTransform

Lift r-cell features to r+1-cells by set operations.

Parameters:
**kwargsoptional

Additional arguments for the class.

__init__(**kwargs)#
forward(data)#

Apply the lifting to the input data.

Parameters:
datatorch_geometric.data.Data | dict

The input data to be lifted.

Returns:
torch_geometric.data.Data | dict

The lifted data.

lift_features(data)#

Concatenate r-cell features to r+1-cell structures.

Parameters:
datatorch_geometric.data.Data | dict

The input data to be lifted.

Returns:
torch_geometric.data.Data | dict

The lifted data.

Submodules#