morphoclass.transforms.node_features.extract_diameters module

Implementation of the branch diameter node feature extractor.

class morphoclass.transforms.node_features.extract_diameters.ExtractDiameters

Bases: morphoclass.transforms.node_features.extract_node_features.ExtractNodeFeatures

Extract the diameter value attached to each node.

extract_node_features(data)

Extract the diameter node features from data.

Parameters

data – A data sample

Returns

Torch tensor of shape (n_nodes,)

Return type

features