morphoclass.transforms.node_features.extract_node_features module

Abstract base class for node feature extractors.

class morphoclass.transforms.node_features.extract_node_features.ExtractNodeFeatures

Bases: abc.ABC

Abstract base class for arbitrary node feature extraction.

abstract extract_node_features(data)

Extract some node features from data.

Parameters

data – A data sample

Returns

Torch tensor of shape (n_nodes, n_features)

Return type

features