Bagging#

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns
from sklearn.model_selection import train_test_split
from sklearn.ensemble import BaggingClassifier
from sklearn.tree import DecisionTreeClassifier
from sklearn.metrics import classification_report, confusion_matrix
import yfinance as yf

Descargar precios del Futuro ES:#

# Descargar datos históricos del futuro del índice S&P 500 (ES)
ticker = "ES=F"
data = yf.download(ticker, start="2020-01-01", end="2024-06-30", interval="1d")

# Graficar el precio de cierre
plt.figure(figsize=(14, 7))
plt.plot(data["Close"], label="Precio de Cierre")
plt.title("Precio de Cierre del Futuro del Índice S&P 500 (ES)")
plt.xlabel("Fecha")
plt.ylabel("Precio de Cierre")
plt.legend()
plt.show()
[*******************100%*********************]  1 of 1 completed
../../../_images/output_3_1.png

Creación de las características:#

# Crear características adicionales
data["PriceChange"] = data["Close"].shift(-1) - data["Close"]
data["Label"] = np.where(data["PriceChange"] > 0, 1, 0)
data["DailyPctChange"] = ((data["Close"] - data["Open"]) / data["Open"]) * 100
data["Range"] = data["High"] - data["Low"]
data["Volatility"] = (data["High"] - data["Low"]) / data["Close"]

# Seleccionar características y etiqueta
features = [
    "Open",
    "High",
    "Low",
    "Close",
    "Volume",
    "Volatility",
    "DailyPctChange",
    "Range",
]
X = data[features]
y = data["Label"]

print("Cantidad precios positivos y negativos:\n", data["Label"].value_counts())
print(X.head())
Cantidad precios positivos y negativos:
 1    602
0    529
Name: Label, dtype: int64
               Open     High      Low    Close   Volume  Volatility  Date
2020-01-02  3237.00  3261.75  3234.25  3259.00  1416241    0.008438
2020-01-03  3261.00  3263.50  3206.75  3235.50  1755057    0.017540
2020-01-06  3220.25  3249.50  3208.75  3243.50  1502748    0.012564
2020-01-07  3243.50  3254.50  3226.00  3235.25  1293494    0.008809
2020-01-08  3231.75  3267.75  3181.00  3260.25  2279138    0.026608

            DailyPctChange  Range
Date
2020-01-02        0.679642  27.50
2020-01-03       -0.781969  56.75
2020-01-06        0.721994  40.75
2020-01-07       -0.254355  28.50
2020-01-08        0.881875  86.75

Bagging:#

Bagging (Bootstrap Aggregating) es una técnica de aprendizaje de conjunto que mejora la precisión y robustez de un modelo al entrenar múltiples instancias del mismo modelo en diferentes subconjuntos aleatorios del conjunto de datos de entrenamiento. Al combinar las predicciones de estos modelos, Bagging reduce la varianza y el riesgo de sobreajuste, promediando las predicciones de cada modelo individual para obtener un resultado más confiable. Este enfoque es particularmente eficaz para estabilizar modelos de alta varianza como los árboles de decisión.

  • base_estimator: Define el modelo base que se usará en el conjunto, en este caso, un DecisionTreeClassifier. Se puede hacer también con LogisticRegression y SVC (Máquinas de Soporte Vectorial).

  • n_estimators: Especifica el número de modelos que se incluirán en el conjunto.

  • max_samples: Indica el tamaño del subconjunto de datos usado para entrenar cada modelo individual, en este caso, el 80% del conjunto de entrenamiento.

  • bootstrap=True: Activa el muestreo con reemplazo (bagging).

  • n_jobs: Permite el uso de múltiples núcleos de CPU para acelerar el entrenamiento.

# Dividir el conjunto de datos en entrenamiento y prueba
X_train, X_test, y_train, y_test = train_test_split(
    X, y, test_size=0.3, random_state=34
)

# Crear un clasificador de Bagging con Árboles de Decisión
bagging_clf = BaggingClassifier(
    base_estimator=DecisionTreeClassifier(),  # Modelo base: Árbol de Decisión
    n_estimators=100,  # Número de modelos en el conjunto
    max_samples=0.8,  # Tamaño de las muestras de entrenamiento
    bootstrap=True,  # Con reemplazo
    n_jobs=-1,  # Usar todos los núcleos de CPU disponibles
    random_state=34,
)

# Entrenar el clasificador de Bagging
bagging_clf.fit(X_train, y_train)

# Realizar predicciones
y_pred = bagging_clf.predict(X_test)

Evaluación del modelo:#

# Calcular las métricas de evaluación
conf_matrix = confusion_matrix(y_test, y_pred)
class_report = classification_report(y_test, y_pred)

# Mostrar las métricas de evaluación
print("Classification Report:\n", class_report)

# Crear un mapa de calor para la matriz de confusión con etiquetas
plt.figure(figsize=(8, 6))
sns.heatmap(
    conf_matrix,
    annot=True,
    fmt="d",
    cmap="Blues",
    xticklabels=["Predicho 0", "Predicho 1"],
    yticklabels=["Real 0", "Real 1"],
)
plt.xlabel("Etiqueta Predicha")
plt.ylabel("Etiqueta Real")
plt.title("Matriz de Confusión")
plt.show()
Classification Report:
               precision    recall  f1-score   support

           0       0.48      0.49      0.49       154
           1       0.57      0.56      0.57       186

    accuracy                           0.53       340
   macro avg       0.53      0.53      0.53       340
weighted avg       0.53      0.53      0.53       340
../../../_images/output_11_1.png

Comparar con un solo Árbol de Decisión:#

# Crear modelo con pre-poda
decision_tree = DecisionTreeClassifier(random_state=34)
decision_tree.fit(X_train, y_train)

# Predicciones
y_pred = decision_tree.predict(X_test)

# Calcular las métricas de evaluación
conf_matrix = confusion_matrix(y_test, y_pred)
class_report = classification_report(y_test, y_pred)

# Mostrar las métricas de evaluación
print("Classification Report:\n", class_report)

# Crear un mapa de calor para la matriz de confusión con etiquetas
plt.figure(figsize=(8, 6))
sns.heatmap(
    conf_matrix,
    annot=True,
    fmt="d",
    cmap="Blues",
    xticklabels=["Predicho 0", "Predicho 1"],
    yticklabels=["Real 0", "Real 1"],
)
plt.xlabel("Etiqueta Predicha")
plt.ylabel("Etiqueta Real")
plt.title("Matriz de Confusión")
plt.show()
Classification Report:
               precision    recall  f1-score   support

           0       0.44      0.48      0.46       154
           1       0.54      0.50      0.52       186

    accuracy                           0.49       340
   macro avg       0.49      0.49      0.49       340
weighted avg       0.49      0.49      0.49       340
../../../_images/output_13_1.png