Python에서 정수가 아닌 문자열 축으로 혼동 행렬을 그리는 방법은 무엇입니까?

Python에서 정수가 아닌 문자열 축을 사용하여 혼동 행렬을 표시하려면 다음 단계를 수행할 수 있습니다.

• 라벨 목록을 만드세요.
• 정오분류표를 만드십시오. confusion_matrix() 사용 분류의 정확도를 계산합니다.
• 3. '~.axes.Axes' 추가 하위 플롯 배열의 일부로 그림에.
• 2D 행렬 또는 배열의 값을 색상으로 구분된 이미지로 표시합니다.
• colorbar() 사용 메소드, ScalarMappable용 컬러바 생성 인스턴스, *매핑 가능*
• 6. set_xticklabels를 사용하여 x 및 y 눈금 레이블 설정 및 set_yticklabels 메소드
• 그림을 표시하려면 show()를 사용하세요. 방법.

예시

from matplotlib import pyplot as plt
from sklearn.metrics import confusion_matrix
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True
labels = ['business', 'health']
cm = confusion_matrix([3], [2])
fig = plt.figure()
ax = fig.add_subplot(111)
cax = ax.matshow(cm)
fig.colorbar(cax)
ax.set_xticklabels([''] + labels)
ax.set_yticklabels([''] + labels)
plt.show()

출력