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) 的類陣列或稀疏矩陣

訓練資料。直接以 Fortran 連續資料傳遞,以避免不必要的記憶體複製。如果 y 是單輸出,則 X 可以是稀疏矩陣。

y形狀為 (n_samples,) 或 (n_samples, n_targets) 的類陣列或稀疏矩陣

目標值。

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 數量。

alphas類陣列,預設值為 None

要計算模型的 alpha 清單。如果為 None,則會自動設定 alpha。

precompute‘auto’、bool 或形狀為 (n_features, n_features) 的類陣列,預設值為 ‘auto’

是否使用預先計算的 Gram 矩陣來加速計算。如果設定為 'auto',則由我們決定。Gram 矩陣也可以作為參數傳遞。

Xy形狀為 (n_features,) 或 (n_features, n_targets) 的類陣列,預設值為 None

可以預先計算的 Xy = np.dot(X.T, y)。僅在預先計算 Gram 矩陣時才有用。

copy_Xbool,預設值為 True

如果為 True,則會複製 X;否則,可能會覆寫 X。

coef_init形狀為 (n_features, ) 的類陣列,預設值為 None

係數的初始值。

verbosebool 或 int,預設值為 False

詳細程度。

return_n_iterbool,預設值為 False

是否傳回迭代次數。

positivebool,預設值為 False

如果設定為 True,則會強制係數為正數。(僅在 y.ndim == 1 時允許)。

check_inputbool,預設值為 True

如果設定為 False,則會跳過輸入驗證檢查(包括提供的 Gram 矩陣)。假設它們由呼叫者處理。

**paramskwargs

傳遞至座標下降求解器的關鍵字引數。

傳回值:
alphas形狀為 (n_alphas,) 的 ndarray

沿著計算模型的路徑的 alpha 值。

coefs形狀為 (n_features, n_alphas) 或 (n_targets, n_features, n_alphas) 的 ndarray

沿著路徑的係數。

dual_gaps形狀為 (n_alphas,) 的 ndarray

每個 alpha 最佳化結束時的對偶間隙。

n_itersint 的清單

座標下降最佳化器達到每個 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...]])