Skip to content

Mixture Distribution¤

distreqx.distributions.MixtureSameFamily ¤

Mixture with components provided from a single vmapped distribution.

__init__(mixture_distribution: Categorical, components_distribution: AbstractDistribution) -> None ¤

Initializes a mixture distribution for components of a shared family.

Arguments:

  • mixture_distribution: Distribution over selecting components.
  • components_distribution: Component distribution.
posterior_marginal(observation: Array) -> Categorical ¤

Generate the posterior distribution given a datapoint.

Arguments:

  • observation: the data point to compute the distribution over

Returns:

The computed categorical distribution

posterior_mode(observation: Array) -> Array ¤

Compute the most likely component a data point falls into.

Arguments:

  • observation: the data point to compute the mode of

Returns:

The computed mode