treeflow.distributions.discrete module

class treeflow.distributions.discrete.FiniteDiscreteDistribution(*args, **kwargs)

Bases: Protocol

Interface for discrete probability distributions with a finite support

property support: Tensor

Values that discrete distribution can take

property normalised_support: Tensor

Support normalised to have a mean of 1 (weighted by probability mass)

property probabilities

Respective probability masses for support

property support_size

Number of elements in support