shrunk_covariance

sklearn.covariance.shrunk_covariance(emp_cov, shrinkage=0.1)

计算对角线收缩的协方差矩阵。

更多信息请参阅 用户指南 。

Parameters:

emp_cov形状为 (…, n_features, n_features) 的类数组

要收缩的协方差矩阵，至少为 2D ndarray。

shrinkagefloat, 默认=0.1

用于计算收缩估计的凸组合中的系数。范围是 [0, 1]。

Returns:

shrunk_cov形状为 (…, n_features, n_features) 的 ndarray

收缩的协方差矩阵。

Notes

正则化（收缩）协方差由以下公式给出:

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

其中 mu = trace(cov) / n_features 。

Examples

>>> import numpy as np
>>> from sklearn.datasets import make_gaussian_quantiles
>>> from sklearn.covariance import empirical_covariance, shrunk_covariance
>>> real_cov = np.array([[.8, .3], [.3, .4]])
>>> rng = np.random.RandomState(0)
>>> X = rng.multivariate_normal(mean=[0, 0], cov=real_cov, size=500)
>>> shrunk_covariance(empirical_covariance(X))
array([[0.73..., 0.25...],
       [0.25..., 0.41...]])