from sklearn.metrics import confusion_matrix conf_mat = confusion_matrix(y_test, y_pred) sns.heatmap(conf_mat, square=True, annot=True, cmap='Blues', fmt='d', cbar=False)