Note ≡

역주 사이킷런 ConfusionMatrixDisplay 클래스를 사용하면 오차 행렬을 쉽게 그릴 수 있습니다. from_estimator 메서드의 첫 번째 매개변수로 훈련된 모델을 전달하고 그다음 테스트 데이터셋의 특성과 타깃 데이터를 전달하면 됩니다.

>>> from sklearn.metrics import ConfusionMatrixDisplay
>>> ConfusionMatrixDisplay.from_estimator(pipe_svc, X_test, y_test)
>>> plt.show()

▲ 그림 6-13 ConfusionMatrixDisplay 클래스로 그린 오차 행렬

confusion_matrix() 함수와 ConfusionMatrixDisplay 클래스는 normalize 매개변수를 사용하여 오차 행렬의 값을 정규화할 수 있습니다. 'true', 'pred'는 각각 행과 열 방향으로 정규화합니다. 'all'로 지정하면 전체 출력 값을 정규화합니다.