zuko.flows.mixture#

Mixture models.

Classes#

GMM

Creates a Gaussian mixture model (GMM).

Descriptions#

class zuko.flows.mixture.GMM(features, context=0, components=2, **kwargs)#

Creates a Gaussian mixture model (GMM).

\[p(X | c) = \sum_{i = 1}^K w_i(c) \, \mathcal{N}(X | \mu_i(c), \Sigma_i(c))\]

Wikipedia

https://wikipedia.org/wiki/Mixture_model#Gaussian_mixture_model

Parameters:

  • features (int) – The number of features.

  • context (int) – The number of context features.

  • components (int) – The number of components \(K\) in the mixture.

  • kwargs – Keyword arguments passed to zuko.nn.MLP.