코드를 실행해서 얻은 배열은 분류기가 테스트 데이터셋에서 만든 에러의 종류를 알려 줍니다. 맷플롯립의 matshow 함수를 사용하여 이 배열을 앞서 보았던 오차 행렬 그림으로 나타내 보겠습니다.
>>> fig, ax = plt.subplots(figsize=(2.5, 2.5))
>>> ax.matshow(confmat, cmap=plt.cm.Blues, alpha=0.3)
>>> for i in range(confmat.shape[0]):
... for j in range(confmat.shape[1]):
... ax.text(x=j, y=i,
... s=confmat[i, j],
... va='center', ha='center')
>>> plt.xlabel('Predicted label')
>>> plt.ylabel('True label')
>>> plt.tight_layout()
>>> plt.show()
그림 6- 10과 같이 오차 행렬 그림에 레이블을 추가하면 이해하기 좀 더 쉽습니다.
▲ 그림 6-10 유방암 데이터셋의 오차 행렬