Regressão Logística

Atualizado 2020-01-29 Leitura 3 min

Resumo

A regressão logística passa uma combinação linear das entradas pela função sigmoide para prever a probabilidade de que o rótulo seja 1.
A saída pertence ao intervalo $[0, 1]$, permitindo definir limiares de decisão de forma flexível e interpretar os coeficientes como contribuições para as log-chances.
O treinamento minimiza a perda de entropia cruzada (equivalentemente maximiza a log-verossimilhança); a regularização L1/L2 evita o sobreajuste do modelo.
A LogisticRegression do scikit-learn lida com pré-processamento, treinamento e visualização da fronteira de decisão com poucas linhas de código.

Intuição #

Este método deve ser interpretado por meio de suas suposições, condições dos dados e como as escolhas de parâmetros afetam a generalização.

Explicação Detalhada #

Formulação Matemática #

A probabilidade da classe 1 dado $\mathbf{x}$ é

$$ P(y=1 \mid \mathbf{x}) = \sigma(\mathbf{w}^\top \mathbf{x} + b) = \frac{1}{1 + \exp\left(-(\mathbf{w}^\top \mathbf{x} + b)\right)}. $$

O aprendizado maximiza a log-verossimilhança

$$ \ell(\mathbf{w}, b) = \sum_{i=1}^{n} \Bigl[ y_i \log p_i + (1 - y_i) \log (1 - p_i) \Bigr], \quad p_i = \sigma(\mathbf{w}^\top \mathbf{x}_i + b), $$

ou equivalentemente minimiza a perda de entropia cruzada negativa. Adicionar regularização L2 evita que os coeficientes explodam, enquanto a regularização L1 pode levar pesos irrelevantes até zero.

Experimentos em Python #

O trecho de código abaixo ajusta a regressão logística a um conjunto de dados sintético bidimensional e visualiza a fronteira de decisão resultante. Tudo — do treinamento à plotagem — cabe em poucas linhas graças ao scikit-learn.

from __future__ import annotations

import japanize_matplotlib
import matplotlib.pyplot as plt
import numpy as np
from matplotlib.colors import ListedColormap
from sklearn.datasets import make_classification
from sklearn.linear_model import LogisticRegression
from sklearn.metrics import accuracy_score

def run_logistic_regression_demo(
    n_samples: int = 300,
    random_state: int = 2,
    label_class0: str = "class 0",
    label_class1: str = "class 1",
    label_boundary: str = "decision boundary",
    title: str = "Logistic regression decision boundary",
) -> dict[str, float]:
    """Train logistic regression on a synthetic 2D dataset and visualise the boundary.

    Args:
        n_samples: Number of samples to generate.
        random_state: Seed for reproducible sampling.
        label_class0: Legend label for class 0.
        label_class1: Legend label for class 1.
        label_boundary: Legend label for the separating line.
        title: Title for the plot.

    Returns:
        Dictionary containing training accuracy and coefficients.
    """
    japanize_matplotlib.japanize()
    X, y = make_classification(
        n_samples=n_samples,
        n_features=2,
        n_redundant=0,
        n_informative=2,
        random_state=random_state,
        n_clusters_per_class=1,
    )

    clf = LogisticRegression()
    clf.fit(X, y)

    accuracy = float(accuracy_score(y, clf.predict(X)))
    coef = clf.coef_[0]
    intercept = float(clf.intercept_[0])

    x1, x2 = X[:, 0], X[:, 1]
    grid_x1, grid_x2 = np.meshgrid(
        np.linspace(x1.min() - 1.0, x1.max() + 1.0, 200),
        np.linspace(x2.min() - 1.0, x2.max() + 1.0, 200),
    )
    grid = np.c_[grid_x1.ravel(), grid_x2.ravel()]
    probs = clf.predict_proba(grid)[:, 1].reshape(grid_x1.shape)

    cmap = ListedColormap(["#aec7e8", "#ffbb78"])
    fig, ax = plt.subplots(figsize=(7, 6))
    contour = ax.contourf(grid_x1, grid_x2, probs, levels=20, cmap=cmap, alpha=0.4)
    ax.contour(grid_x1, grid_x2, probs, levels=[0.5], colors="k", linewidths=1.5)
    ax.scatter(x1[y == 0], x2[y == 0], marker="o", edgecolor="k", label=label_class0)
    ax.scatter(x1[y == 1], x2[y == 1], marker="x", color="k", label=label_class1)
    ax.set_xlabel("feature 1")
    ax.set_ylabel("feature 2")
    ax.set_title(title)
    ax.legend(loc="best")
    fig.colorbar(contour, ax=ax, label="P(class = 1)")
    fig.tight_layout()
    plt.show()

    return {
        "accuracy": accuracy,
        "coef_0": float(coef[0]),
        "coef_1": float(coef[1]),
        "intercept": intercept,
    }

metrics = run_logistic_regression_demo(
    label_class0="class 0",
    label_class1="class 1",
    label_boundary="decision boundary",
    title="Logistic regression decision boundary",
)
print(f"Training accuracy: {metrics['accuracy']:.3f}")
print(f"Coefficient feature 1: {metrics['coef_0']:.3f}")
print(f"Coefficient feature 2: {metrics['coef_1']:.3f}")
print(f"Intercept: {metrics['intercept']:.3f}")

Tudo — do treinamento à plotagem — cabe em poucas linhas graças ao scikit-learn figura

Referências #

Agresti, A. (2015). Foundations of Linear and Generalized Linear Models. Wiley.
Hastie, T., Tibshirani, R., & Friedman, J. (2009). The Elements of Statistical Learning. Springer.