ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.from_random

classmethod GaussianMixture.from_random(key, n_components, n_dimensions, stdev_mean=0.1, stdev_cov=0.1, stdev_weights=0.1, ridge=0, dtype=None)

Construct a random GMM.

Return type

GaussianMixture

Parameters

• key (jax.Array) –

• n_components (int) –

• n_dimensions (int) –

• stdev_mean (float) –

• stdev_cov (float) –

• stdev_weights (float) –

• ridge (Union[float, jax._src.numpy.lax_numpy.array]) –

• dtype (Optional[numpy.dtype]) –