ott.tools.gaussian_mixture.fit_gmm.initialize#
- ott.tools.gaussian_mixture.fit_gmm.initialize(rng, points, point_weights, n_components, n_attempts=50, verbose=False)[source]#
Initialize a GMM via K-means++ with retries on failure.
- Parameters:
rng (
Array
) – jax.random keypoints (
Array
) – (n, n_dimensions) array of observationspoint_weights (
Optional
[Array
]) – (n,) array of weights for pointsn_components (
int
) – desired number of componentsn_attempts (
int
) – number of attempts to initialize before failingverbose (
bool
) – if True, print status information
- Return type:
- Returns:
An initial Gaussian mixture model.
- Raises:
ValueError if initialization was unsuccessful after n_attempts attempts. –