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