[ Foro de Python ]

Mapa de calor

08-Apr-2021 13:46
Invitado (Antonio)
0 Respuestas

Hola buenas, al ejecutar el siguiente código se me genera un mapa de calor con 6 categorías, sin embargo, en hol guardo 11 categorías no sé por qué pasa esto.
def plot_cm(y_true, y_pred, hol):
 cm = confusion_matrix(y_true, y_pred)
 fig, ax = plt.subplots(figsize=(30, 16))
 ax = sns.heatmap(
     cm,
     annot=True, #escribe dentro de cada cuadrado
     fmt="d", #anotaciones
     cmap=sns.diverging_palette(220, 30, n=10),#colores
     ax=ax #ejes
 )

 plt.ylabel('Actual')
 plt.xlabel('Predicted')
 ax.set_xticklabels(hol) #establecemos las etiquetas eje x
 ax.set_yticklabels(hol) #establecemos las etiquetas eje y
 plt.show() #graficamos

hal=enc.inverse_transform(y_test)
hel=enc.inverse_transform(y_pred)
hol=enc.categories_[0]
plot_cm(
 hal, #label eje y
 hel, #label eje x
 hol
)




(No se puede continuar esta discusión porque tiene más de dos meses de antigüedad. Si tienes dudas parecidas, abre un nuevo hilo.)