จะพล็อตเมทริกซ์ความสับสนด้วยแกนสตริงแทนที่จะเป็นจำนวนเต็มใน Python ได้อย่างไร

ในการพล็อตเมทริกซ์ความสับสนด้วยแกนสตริงแทนที่จะเป็นจำนวนเต็มใน Python เราสามารถทำตามขั้นตอนต่อไปนี้

• สร้างรายการสำหรับป้ายกำกับ
• สร้างเมทริกซ์ความสับสน ใช้ confusion_matrix() เพื่อคำนวณความถูกต้องของการจำแนกประเภท
• 3. เพิ่ม '~.axes.Axes' ให้กับร่างซึ่งเป็นส่วนหนึ่งของการจัดโครงเรื่องย่อย
• พล็อตค่าของเมทริกซ์หรืออาร์เรย์ 2 มิติเป็นรูปภาพที่มีรหัสสี
• การใช้ แถบสี() เมธอด สร้างแถบสีสำหรับ ScalarMappable ตัวอย่าง *แมปได้*
• 6. ตั้งค่า x และ y ticklabels โดยใช้ set_xticklabels และ เมธอด 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()

ผลลัพธ์