코드를 실행해서 얻은 배열은 분류기가 테스트 데이터셋에서 만든 에러의 종류를 알려 줍니다. 맷플롯립의 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 유방암 데이터셋의 오차 행렬

    신간 소식 구독하기
    뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.