L1-逻辑回归的正则化路径#

在源自Iris数据集的二元分类问题上训练L1惩罚的逻辑回归模型。

模型按正则化强度从强到弱排序。收集模型的4个系数并绘制为“正则化路径”:在图表的左侧(强正则化器),所有系数都精确为0。当正则化逐渐放松时,系数可以依次获得非零值。

这里我们选择liblinear求解器,因为它能有效地优化具有非平滑、诱导稀疏性的L1惩罚的逻辑回归损失。

另请注意,我们设置了较低的容差值,以确保模型在收集系数之前已收敛。

我们还使用warm_start=True,这意味着模型的系数被重用于初始化下一次模型拟合,以加快计算完整路径的速度。

# Authors: The scikit-learn developers
# SPDX-License-Identifier: BSD-3-Clause

加载数据#

from sklearn import datasets

iris = datasets.load_iris()
X = iris.data
y = iris.target
feature_names = iris.feature_names

这里我们移除第三类,使问题成为二元分类

X = X[y != 2]
y = y[y != 2]

计算正则化路径#

import numpy as np

from sklearn.linear_model import LogisticRegression
from sklearn.pipeline import make_pipeline
from sklearn.preprocessing import StandardScaler
from sklearn.svm import l1_min_c

cs = l1_min_c(X, y, loss="log") * np.logspace(0, 1, 16)

使用StandardScalerLogisticRegression创建一个管道,在拟合线性模型之前对数据进行归一化,以加快收敛速度并使系数具有可比性。此外,作为一个附带作用,由于数据现在以0为中心,我们不需要拟合截距。

clf = make_pipeline(
    StandardScaler(),
    LogisticRegression(
        l1_ratio=1,
        solver="liblinear",
        tol=1e-6,
        max_iter=int(1e6),
        warm_start=True,
        fit_intercept=False,
    ),
)
coefs_ = []
for c in cs:
    clf.set_params(logisticregression__C=c)
    clf.fit(X, y)
    coefs_.append(clf["logisticregression"].coef_.ravel().copy())

coefs_ = np.array(coefs_)

绘制正则化路径#

import matplotlib.pyplot as plt

# Colorblind-friendly palette (IBM Color Blind Safe palette)
colors = ["#648FFF", "#785EF0", "#DC267F", "#FE6100"]

plt.figure(figsize=(10, 6))
for i in range(coefs_.shape[1]):
    plt.semilogx(cs, coefs_[:, i], marker="o", color=colors[i], label=feature_names[i])

ymin, ymax = plt.ylim()
plt.xlabel("C")
plt.ylabel("Coefficients")
plt.title("Logistic Regression Path")
plt.legend()
plt.axis("tight")
plt.show()
Logistic Regression Path

脚本总运行时间: (0 minutes 0.138 seconds)

相关示例

岭系数作为 L2 正则化的函数

岭系数作为 L2 正则化的函数

模型正则化对训练和测试误差的影响

模型正则化对训练和测试误差的影响

绘制岭系数作为正则化函数的函数

绘制岭系数作为正则化函数的函数

逻辑回归中的 L1 惩罚和稀疏性

逻辑回归中的 L1 惩罚和稀疏性

由 Sphinx-Gallery 生成的图库