Implementando a regressão linear em Python

Agora que você entende o conceito de regressão linear, vamos implementá-la usando Python. Usaremos a popular biblioteca scikit-learn para isso.

Etapa 1: preparando os dados

Primeiro, precisamos preparar um conjunto de dados. Neste exemplo, usaremos um conjunto simples com uma variável independente (x) e uma variável dependente (y).

Aqui estão os dados:

• x: 1, 2, 3, 4, 5
• y: 2, 4, 6, 8, 10

Etapa 2: importando as bibliotecas necessárias

Precisamos importar as seguintes bibliotecas:

• numpy: para cálculos numéricos.
• matplotlib: para visualizações.
• scikit-learn: para algoritmos de aprendizado de máquina.

import numpy as np
import matplotlib.pyplot as plt
from sklearn.linear_model import LinearRegression

Etapa 3: criando o modelo

Criamos um modelo de regressão linear usando a classe LinearRegression do scikit-learn e ajustamos o modelo aos dados.

x = np.array([1, 2, 3, 4, 5]).reshape(-1, 1)
y = np.array([2, 4, 6, 8, 10])

model = LinearRegression()
model.fit(x, y)

Etapa 4: fazendo previsões

Depois que o modelo é treinado, podemos usá-lo para fazer previsões. Por exemplo, para prever o valor de y quando x = 6.

prediction = model.predict(np.array([[6]]))
print(f"Predicted value for x=6: {prediction[0]}")

Etapa 5: visualizando os resultados

Podemos plotar os pontos de dados e a linha de regressão para visualizar o quanto o modelo se ajusta aos dados.

plt.scatter(x, y, color='blue', label='Data points')
plt.plot(x, model.predict(x), color='red', label='Regression line')
plt.title('Linear Regression Visualization')
plt.xlabel('X')
plt.ylabel('Y')
plt.legend()
plt.show()

Etapa 6: avaliando o modelo

Avaliamos o desempenho do modelo usando métricas como R-quadrado. Essa métrica mede o quanto o modelo explica a variabilidade dos dados.

r_squared = model.score(x, y)
print(f"R-squared value: {r_squared}")

Ao implementar a regressão linear, você pode enfrentar desafios como:

• Sobreajuste com conjuntos de dados pequenos.
• Sensibilidade a valores discrepantes.
• Como lidar com multicolinearidade em várias variáveis.

Nesta lição, nós:

• Aprendemos a implementar a regressão linear usando scikit-learn.
• Fizemos previsões e visualizamos os resultados.
• Avaliamos o modelo usando R-quadrado.

Em seguida, exploraremos o conceito de regressão logística para problemas de classificação.