topobench.transforms.data_manipulations.keep_selected_data_fields module

A transform that keeps only the selected fields of the input data.

class KeepSelectedDataFields(**kwargs)

Bases: BaseTransform

A transform that keeps only the selected fields of 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 transformed data.