Случайные белые линии в матрице путаницы - PullRequest
0 голосов
/ 02 мая 2020

Ниже приведено изображение матрицы путаницы. Каждый раз, когда я строю матрицу, я получаю случайные белые линии после нормализации матрицы.

_pred = np.array(preds)
_y_test = np.array(y_test.argmax(axis=-1))

cm = confusion_matrix(_pred, _y_test)
cmn = cm.astype('float') / cm.sum(axis=1)[:, np.newaxis]

labels = list(range(1, num_subj))
df_cm = pd.DataFrame(cmn, range(1, num_subj), range(1, num_subj))
sn.heatmap(df_cm, annot=True, annot_kws={"size": 6}, fmt='.2f')
plt.ylabel('Actual')
plt.xlabel('Predicted')
plt.show()

Как от этого избавиться? Это происходит только при нормализации матрицы.

confusion_matrix

...