topobench.transforms.data_manipulations.keep_selected_target_indices module

A transform that keeps only particular target field indices.

class KeepSelectedTargetIndices(**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.