topobench.transforms.data_manipulations.node_features_to_float module

A transform that converts the node features of the input graph to float.

class NodeFeaturesToFloat(**kwargs)

Bases: BaseTransform

A transform that converts the node features of the input graph to float.

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.