orthogonal_mp_gram#
- sklearn.linear_model.orthogonal_mp_gram(Gram, Xy, *, n_nonzero_coefs=None, tol=None, norms_squared=None, copy_Gram=True, copy_Xy=True, return_path=False, return_n_iter=False)[source]#
Gram 正交匹配追踪(OMP)。
仅使用 Gram 矩阵 X.T * X 和乘积 X.T * y 解决 n_targets 个正交匹配追踪问题。
在用户指南中了解更多。
- 参数:
- Gram形状为 (n_features, n_features) 的类数组
输入数据的 Gram 矩阵:
X.T * X
。- Xy形状为 (n_features,) 或 (n_features, n_targets) 的类数组
输入目标乘以
X
:X.T * y
。- n_nonzero_coefs整型,默认为 None
解中非零项的所需数量。如果为
None
(默认),则此值设置为 n_features 的 10%。- tol浮点型,默认为 None
残差的最大平方范数。如果不是
None
,则会覆盖n_nonzero_coefs
。- norms_squared形状为 (n_targets,) 的类数组,默认为 None
y
各行的 L2 范数平方。如果tol
不为 None,则必需。- copy_Gram布尔型,默认为 True
算法是否必须复制 Gram 矩阵。只有当 Gram 矩阵已经是 Fortran 顺序时,
False
值才有用,否则无论如何都会进行复制。- copy_Xy布尔型,默认为 True
算法是否必须复制协方差向量
Xy
。如果为False
,则可能会被覆盖。- return_path布尔型,默认为 False
是否返回前向路径中非零系数的每个值。对交叉验证有用。
- return_n_iter布尔型,默认为 False
是否返回迭代次数。
- 返回:
- coef形状为 (n_features,) 或 (n_features, n_targets) 的 ndarray
OMP 解的系数。如果
return_path=True
,则包含整个系数路径。在这种情况下,其形状为(n_features, n_features)
或(n_features, n_targets, n_features)
,并且对最后一个轴进行迭代会按活动特征的递增顺序生成系数。- n_iters列表或整型
每个目标中的活动特征数量。仅当
return_n_iter
设置为 True 时返回。
另请参阅
OrthogonalMatchingPursuit
正交匹配追踪模型(OMP)。
orthogonal_mp
解决 n_targets 个正交匹配追踪问题。
lars_path
使用 LARS 算法计算最小角回归或 Lasso 路径。
sklearn.decomposition.sparse_encode
通用稀疏编码。结果的每一列都是 Lasso 问题的解。
注释
正交匹配追踪由 G. Mallat 和 Z. Zhang 在 IEEE Transactions on Signal Processing, Vol. 41, No. 12. (December 1993), pp. 3397-3415 发表的论文《Matching pursuits with time-frequency dictionaries》中提出。 (https://www.di.ens.fr/~mallat/papiers/MallatPursuit93.pdf)
此实现基于 Rubinstein, R., Zibulevsky, M. 和 Elad, M. 于 2008 年 4 月发表的《Efficient Implementation of the K-SVD Algorithm using Batch Orthogonal Matching Pursuit Technical Report - CS Technion》。 https://www.cs.technion.ac.il/~ronrubin/Publications/KSVD-OMP-v2.pdf
示例
>>> from sklearn.datasets import make_regression >>> from sklearn.linear_model import orthogonal_mp_gram >>> X, y = make_regression(noise=4, random_state=0) >>> coef = orthogonal_mp_gram(X.T @ X, X.T @ y) >>> coef.shape (100,) >>> X[:1,] @ coef array([-78.68])