Làm thế nào để vẽ một ma trận nhầm lẫn với trục chuỗi thay vì số nguyên trong Python?

Để vẽ một ma trận nhầm lẫn với trục chuỗi thay vì số nguyên trong Python, chúng ta có thể thực hiện các bước sau đây−

• Lập danh sách cho các nhãn.
• Tạo một ma trận nhầm lẫn. Sử dụng nhầm lẫn_matrix () để tính toán độ chính xác của phân loại.
• 3. Thêm '~ .axes.Axes' vào hình như một phần của sắp xếp ô phụ.
• Vẽ đồ thị các giá trị của ma trận hoặc mảng 2D dưới dạng hình ảnh được mã hóa màu.
• Sử dụng colorbar () , tạo thanh màu cho ScalarMappable ví dụ, * có thể lập bản đồ *
• 6. Đặt các nhãn đánh dấu x và y bằng cách sử dụng set_xticklabels và phương thức set_yticklabels.
• Để hiển thị hình này, hãy sử dụng show () phương pháp.

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

Đầu ra