sklearn.covariance.shrunk_covariance

sklearn.covariance.shrunk_covariance(emp_cov, shrinkage=0.1)[source]

Calculates a covariance matrix shrunk on the diagonal

Read more in the User Guide.

Parameters
emp_covarray-like, shape (n_features, n_features)

Covariance matrix to be shrunk

shrinkagefloat, 0 <= shrinkage <= 1

Coefficient in the convex combination used for the computation of the shrunk estimate.

Returns
shrunk_covarray-like

Shrunk covariance.

Notes

The regularized (shrunk) covariance is given by:

(1 - shrinkage) * cov + shrinkage * mu * np.identity(n_features)

where mu = trace(cov) / n_features