Bản đồ Nhiệt Ma trận Nhầm lẫn (Confusion Matrix Heatmap) có Chú thích Ô

Mục tiêu kiến thức

• Vẽ ma trận 2D dưới dạng bản đồ nhiệt bằng ax.imshow().
• Tự động ghi nhãn số đếm trực tiếp lên từng ô của ma trận bằng ax.text().
• Thiết lập nhãn trục tương ứng với tên các lớp dự đoán và nhãn thực tế.

Mô tả bài toán

Viết hàm plot_confusion_matrix_heatmap(cm: list[list[int]], class_names: list[str], cmap: str = "Blues") -> matplotlib.figure.Figure:

• cm: ma trận nhầm lẫn kích thước N × N.
• Tạo fig, ax = plt.subplots(figsize=(6, 6)).
• Hiển thị ma trận: im = ax.imshow(cm, cmap=cmap).
• Thêm thanh chỉ thị màu: fig.colorbar(im, ax=ax).
• Đặt các mốc ticks:
• ax.set_xticks(range(len(class_names))), nhãn class_names.
• ax.set_yticks(range(len(class_names))), nhãn class_names.
• Với mỗi ô (i, j) trong ma trận:
• Điền giá trị số cm[i][j] vào vị trí tương ứng bằng ax.text(j, i, str(cm[i][j]), ha="center", va="center").
• Đặt nhãn: ax.set_xlabel("Predicted Label"), ax.set_ylabel("True Label").
• Đặt tiêu đề: "Confusion Matrix".
• Trả về fig.

Ràng buộc

• Thời gian chạy tối đa: 1000ms.
• Giới hạn bộ nhớ: 256MB.
• Dữ liệu đầu vào tuân thủ đúng kiểu dữ liệu và miền giá trị được mô tả.