Matriz de Confusión con diferentes colores

Answer

1. 

   davy.ai

   June 3, 2023 at 12:26 am

   Sí, hay una forma de agregar etiquetas y cambiar el esquema de colores en seaborn. Puedes agregar etiquetas personalizadas usando el parámetro annot en la función sns.heatmap(). Para cambiar los colores, puedes crear un mapa de colores personalizado utilizando la función ListedColormap() del módulo matplotlib.colors.

   Aquí te mostramos cómo puedes modificar el código existente para lograr el resultado deseado:

   import numpy as np
   import matplotlib.pyplot as plt
   import seaborn as sns
   from matplotlib.colors import ListedColormap
   
   # Define la matriz de confusión.
   cm = np.array([[2547, 288], [255, 216]])
   
   # Define el mapa de colores.
   cmap = ListedColormap(['#FF4136', '#2ECC40'])
   
   # Define las etiquetas.
   labels = [['Verdadero Positivo', 'Falso Negativo'], 
             ['Falso Positivo', 'Verdadero Negativo']]
   
   # Dibuja la matriz de confusión con colores y etiquetas personalizadas.
   fig, ax = plt.subplots(figsize=(8, 6))
   sns.heatmap(cm, annot=True, fmt='g', cmap=cmap, square=True,
               xticklabels=['Predicho Falso', 'Predicho Verdadero'],
               yticklabels=['Actual Falso', 'Actual Verdadero'], ax=ax)
   
   # Agrega las etiquetas al gráfico.
   for i in range(len(labels)):
       for j in range(len(labels)):
           text = ax.text(j+0.5, i+0.5, labels[i][j],
                          ha='center', va='center', color='black')
   
   # Agrega un título al gráfico.
   plt.title('Matriz de confusión personalizada de Seaborn')
   
   # Muestra el gráfico.
   plt.show()
   

   Este código produce una matriz de confusión que se parece a la segunda imagen que proporcionaste.

   Ten en cuenta que en el código anterior, definimos un mapa de colores personalizado utilizando la función ListedColormap(). Pasamos una lista de dos colores a esta función, uno para las predicciones correctas y otro para las incorrectas. También definimos una lista de etiquetas para las diferentes celdas en la matriz de confusión, que agregamos al gráfico utilizando la función text().

   También hicimos algunos otros cambios en el gráfico, como hacerlo cuadrado y cambiar el tamaño de la figura. Puedes modificar estos parámetros según sea necesario para obtener el resultado deseado.

Comments are closed.