topobench.transforms.data_manipulations.node_degrees module#
Node degrees transform.
- class NodeDegrees(**kwargs)#
Bases:
BaseTransformA transform that calculates the node degrees of the input graph.
- Parameters:
- **kwargsoptional
Parameters for the base transform.
- __init__(**kwargs)#
- calculate_node_degrees(data, field)#
Calculate the node degrees of the input data.
- Parameters:
- datatorch_geometric.data.Data
The input data.
- fieldstr
The field to calculate the node degrees.
- Returns:
- torch_geometric.data.Data
The transformed data.
- 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.