Skip to content

Mixture Distribution¤

distreqx.distributions.MixtureSameFamily(distreqx.distributions.AbstractSTDDistribution, distreqx.distributions.AbstractSampleLogProbDistribution, distreqx.distributions.AbstractSurvivalDistribution, distreqx.distributions.AbstractProbDistribution, distreqx.distributions.AbstractCDFDistribution) ¤

Mixture with components provided from a single vmapped distribution.

__init__(mixture_distribution: distreqx.distributions.Categorical, components_distribution: distreqx.distributions.AbstractDistribution) ¤

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) -> distreqx.distributions.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