topobench.transforms.data_manipulations.rename_fields module#
RenameFields transform that does nothing to the input data.
- class RenameFields(init_field_name, new_field_name, **kwargs)#
Bases:
BaseTransformA transform that renames specified fields in a torch_geometric.data.Data object.
- Parameters:
- init_field_namelist of str
List of original field names to be renamed.
- new_field_namelist of str
List of new field names corresponding to init_field_name.
- **kwargsdict, optional
Additional keyword arguments stored on the transform as
self.parameters.
- __init__(init_field_name, new_field_name, **kwargs)#
- forward(data)#
Apply the transform to rename fields in the input data.
- Parameters:
- datatorch_geometric.data.Data
The input data.
- Returns:
- torch_geometric.data.Data
The modified data with renamed fields.