弹性网络路径 (enet_path)#
- sklearn.linear_model.enet_path(X, y, *, l1_ratio=0.5, eps=0.001, n_alphas=100, alphas=None, precompute='auto', Xy=None, copy_X=True, coef_init=None, verbose=False, return_n_iter=False, positive=False, check_input=True, **params)[source]#
使用坐标下降法计算弹性网络路径。
弹性网络优化函数对于单输出和多输出任务有所不同。
对于单输出任务,其公式为:
1 / (2 * n_samples) * ||y - Xw||^2_2 + alpha * l1_ratio * ||w||_1 + 0.5 * alpha * (1 - l1_ratio) * ||w||^2_2
对于多输出任务,其公式为:
(1 / (2 * n_samples)) * ||Y - XW||_Fro^2 + alpha * l1_ratio * ||W||_21 + 0.5 * alpha * (1 - l1_ratio) * ||W||_Fro^2
其中
||W||_21 = \sum_i \sqrt{\sum_j w_{ij}^2}
即每行的范数之和。
更多信息请参见 用户指南。
- 参数:
- X形状为 (n_samples, n_features) 的 {array-like, sparse matrix}
训练数据。为了避免不必要的内存复制,请直接作为 Fortran 连续数据传递。如果
y
是单输出,则X
可以是稀疏矩阵。- y形状为 (n_samples,) 或 (n_samples, n_targets) 的 {array-like, sparse matrix}
目标值。
- l1_ratiofloat, 默认值=0.5
传递给弹性网络的 0 到 1 之间的数字(l1 和 l2 惩罚之间的缩放)。
l1_ratio=1
对应于 Lasso。- epsfloat, 默认值=1e-3
路径长度。
eps=1e-3
表示alpha_min / alpha_max = 1e-3
。- n_alphasint, 默认值=100
正则化路径上的 alpha 数量。
- alphasarray-like, 默认值=None
计算模型的 alpha 列表。如果为 None,则自动设置 alpha。
- precompute‘auto’,bool 或形状为 (n_features, n_features) 的 array-like,默认值='auto'
是否使用预计算的 Gram 矩阵来加速计算。如果设置为
'auto'
,则由程序自动决定。Gram 矩阵也可以作为参数传递。- Xy形状为 (n_features,) 或 (n_features, n_targets) 的 array-like,默认值=None
Xy = np.dot(X.T, y) 可以预先计算。只有在预计算 Gram 矩阵时才有用。
- copy_Xbool, 默认值=True
如果为
True
,则复制 X;否则,可能会覆盖 X。- coef_init形状为 (n_features,) 的 array-like,默认值=None
系数的初始值。
- verbosebool 或 int,默认值=False
冗余信息的多少。
- return_n_iterbool, 默认值=False
是否返回迭代次数。
- positivebool, 默认值=False
如果设置为 True,则强制系数为正。(仅当
y.ndim == 1
时允许)。- check_inputbool, 默认值=True
如果设置为 False,则跳过输入验证检查(包括提供的 Gram 矩阵)。假设这些检查由调用者处理。
- **paramskwargs
传递给坐标下降求解器的关键字参数。
- 返回值:
- alphas形状为 (n_alphas,) 的 ndarray
计算模型的路径上的 alphas。
- coefs形状为 (n_features, n_alphas) 或 (n_targets, n_features, n_alphas) 的 ndarray
路径上的系数。
- dual_gaps形状为 (n_alphas,) 的 ndarray
每个 alpha 优化结束时的对偶间隙。
- n_iters整数列表
坐标下降优化器为每个 alpha 达到指定容差所花费的迭代次数。(当
return_n_iter
设置为 True 时返回)。
另请参见
MultiTaskElasticNet
使用 L1/L2 混合范数作为正则化器的多任务弹性网络模型。
MultiTaskElasticNetCV
具有内置交叉验证的多任务 L1/L2 弹性网络。
ElasticNet
使用组合的 L1 和 L2 先验作为正则化器的线性回归。
ElasticNetCV
沿正则化路径进行迭代拟合的弹性网络模型。
备注
示例请参见 examples/linear_model/plot_lasso_lasso_lars_elasticnet_path.py。
示例
>>> from sklearn.linear_model import enet_path >>> from sklearn.datasets import make_regression >>> X, y, true_coef = make_regression( ... n_samples=100, n_features=5, n_informative=2, coef=True, random_state=0 ... ) >>> true_coef array([ 0. , 0. , 0. , 97.9..., 45.7...]) >>> alphas, estimated_coef, _ = enet_path(X, y, n_alphas=3) >>> alphas.shape (3,) >>> estimated_coef array([[ 0. , 0.78..., 0.56...], [ 0. , 1.12..., 0.61...], [-0. , -2.12..., -1.12...], [ 0. , 23.04..., 88.93...], [ 0. , 10.63..., 41.56...]])