ott.tools.gaussian_mixture.gaussian_mixture.GaussianMixture.from_points_and_assignment_probs#

classmethod GaussianMixture.from_points_and_assignment_probs(points, point_weights, assignment_probs)[source]#

Estimate a GMM from points and a set of component probabilities.

Return type

GaussianMixture

Parameters

  • points (jax.Array) –

  • point_weights (jax.Array) –

  • assignment_probs (jax.Array) –