पायथन में पूर्णांक के बजाय स्ट्रिंग अक्ष के साथ एक भ्रम मैट्रिक्स को प्लॉट करने के लिए, हम निम्नलिखित कदम उठा सकते हैं-
- लेबल के लिए एक सूची बनाएं।
- भ्रम मैट्रिक्स बनाएँ। confusion_matrix() का उपयोग करें वर्गीकरण की सटीकता की गणना करने के लिए।
- 3. एक '~.axes.Axes' Add जोड़ें एक सबप्लॉट व्यवस्था के हिस्से के रूप में चित्र के लिए।
- 2D मैट्रिक्स या सरणी के मानों को रंग-कोडित छवि के रूप में प्लॉट करें।
- कलरबार () का उपयोग करना विधि, ScalarMappable . के लिए एक कलरबार बनाएं उदाहरण, *मैप करने योग्य*
- 6. set_xticklabels . का उपयोग करके x और y टिकलेबल सेट करें और set_yticlabels विधियां।
- आंकड़ा प्रदर्शित करने के लिए, दिखाएं () . का उपयोग करें विधि।
उदाहरण
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()
आउटपुट