confusion matrix

from sklearn.metrics import confusion_matrix

cm = confusion_matrix(y_true, y_pred)

다음과 같은 방법으로 confusion matrix를 구할 수 있다. 이 때 label은 labelencoding 한 숫자로 변환되는데 이를 다시 decoding 가능하다.

labels = ['ball', 'cage', 'eccen', 'normal']
cm_display = ConfusionMatrixDisplay(confusion_matrix=cm, display_labels=labels)

fig, ax = plt.subplots(figsize=(8, 8))
cm_display.plot(ax=ax)

위와 같은 방법으로 변환 가능하다.

Seaborn heatmap에서 parameter는 다음과 같다.