Cách vẽ ma trận nhầm lẫn với nhãn chuỗi thay vì số nguyên trong Python

Để vẽ ma trận nhầm lẫn với nhãn chuỗi:

• Tạo danh sách nhãn.
• Tạo ma trận nhầm lẫn. Dùng confusion_matrix() tính độ chính xác phân loại.
• Thêm trục vào hình.
• Vẽ giá trị ma trận 2D dạng ảnh màu.
• Tạo colorbar bằng colorbar() cho ScalarMappable.
• Đặt nhãn tick x, y bằng set_xticklabels và set_yticklabels.
• Hiển thị bằng show().

Ví dụ

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()

Kết quả