Implementación de la regresión lineal en Python

Ahora que comprende el concepto de regresión lineal, vamos a implementarla utilizando Python. Para ello, usaremos la popular biblioteca scikit-learn.

Paso 1: Preparar los datos

Primero, debemos preparar un conjunto de datos. En este ejemplo, utilizaremos un conjunto de datos sencillo con una variable independiente (x) y una variable dependiente (y).

Estos son los datos:

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

Paso 2: Importar las bibliotecas necesarias

Debemos importar las siguientes bibliotecas:

• numpy: Para realizar cálculos numéricos.
• matplotlib: Para crear visualizaciones.
• scikit-learn: Para utilizar algoritmos de machine learning.

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

Paso 3: Crear el modelo

Creamos un modelo de regresión lineal utilizando la clase LinearRegression de scikit-learn y lo ajustamos a los datos.

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)

Paso 4: Realizar predicciones

Una vez entrenado el modelo, podemos utilizarlo para realizar predicciones. Por ejemplo, para predecir el valor de y cuando x = 6.

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

Paso 5: Visualizar los resultados

Podemos representar los puntos de datos y la línea de regresión para visualizar hasta qué punto el modelo se ajusta a los datos.

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()

Paso 6: Evaluar el modelo

Evaluamos el rendimiento del modelo mediante métricas como R-squared. Esta métrica mide hasta qué punto el modelo explica la variabilidad de los datos.

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

Al implementar la regresión lineal, puede enfrentarse a dificultades como:

• Sobreajuste con conjuntos de datos pequeños.
• Sensibilidad a los valores atípicos.
• Gestión de la multicolinealidad en varias variables.

En esta lección hemos:

• Aprendido a implementar la regresión lineal utilizando scikit-learn.
• Realizado predicciones y visualizado los resultados.
• Evaluado el modelo mediante R-squared.

A continuación, exploraremos el concepto de regresión logística para problemas de clasificación.