ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.from_mean_cov_component_weights#

classmethod GaussianMixture.from_mean_cov_component_weights(mean, cov, component_weights)[source]#

Construct a GMM from means, covariances, and component weights.

Parameters

  • mean (jax.Array) –

  • cov (jax.Array) –

  • component_weights (jax.Array) –