[ Foro de Python ]
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.)