GaussianMixture.fit

GaussianMixture.fit(data)[source]

Fits the Gaussian mixture on the provided data, estimating component priors, means and covariances. Parameters are estimated using the EM algorithm.

Parameters

data (Union[ndarray[Any, dtype[float32]], Tensor]) -- The tabular data to fit on. The dimensionality of the Gaussian mixture is automatically inferred from this data.

Return type

GaussianMixture

Returns

The fitted Gaussian mixture.