morphoclass.transforms.augmentors.apply_node_reduction_masks module

Implementation of the ApplyNodeReductionMasks transform.

class morphoclass.transforms.augmentors.apply_node_reduction_masks.ApplyNodeReductionMasks

Bases: object

Transform that applies the apical node masks to the apicals.

static apply_mask(apical, mask)

Apply a node mask to the given apical.

Parameters
  • apical (tmd.Tree.Tree) – The apical dendrite tree.

  • mask (np.ndarray) – The node mask.