topobench.transforms.data_manipulations.equal_gaus_features module

This module contains a transform that generates equal Gaussian features for all nodes in the input graph.

class EqualGausFeatures(**kwargs)

Bases: BaseTransform

A transform that generates equal Gaussian features for all nodes.

Parameters:

**kwargsoptional

Additional arguments for the class. It should contain the following keys: - mean (float): The mean of the Gaussian distribution. - std (float): The standard deviation of the Gaussian distribution. - num_features (int): The number of features to generate.

__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.