morphoclass.transforms.node_features.extract_const_feature module

Implementation of the node feature extractor for constant node features.

class morphoclass.transforms.node_features.extract_const_feature.ExtractConstFeature

Bases: morphoclass.transforms.node_features.extract_node_features.ExtractNodeFeatures

Extractor that assigns the constant feature value 1.0 to each node.

extract_node_features(data)

Extract the constant node feature value 1.0 for each node.

Parameters

data – A data sample

Returns

Torch tensor of shape (n_nodes,)

Return type

features