topobench.transforms.data_manipulations.calculate_simplicial_curvature module

A transform that calculates the simplicial curvature of the input graph.

class CalculateSimplicialCurvature(**kwargs)

Bases: BaseTransform

A transform that calculates the simplicial curvature of the input graph.

Parameters:

**kwargsoptional

Parameters for the 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 transformed data.

one_cell_curvature(data)

Calculate the one cell curvature of the input data.

Parameters:

datatorch_geometric.data.Data

The input data.

Returns:

torch_geometric.data.Data

Data with the one cell curvature.

two_cell_curvature(data)

Calculate the two cell curvature of the input data.

Parameters:

datatorch_geometric.data.Data

The input data.

Returns:

torch_geometric.data.Data

Data with the two cell curvature.

zero_cell_curvature(data)

Calculate the zero cell curvature of the input data.

Parameters:

datatorch_geometric.data.Data

The input data.

Returns:

torch_geometric.data.Data

Data with the zero cell curvature.