混淆矩阵与轴的不同标签

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了混淆矩阵与轴的不同标签相关的知识,希望对你有一定的参考价值。

我试图训练和测试朴素贝叶斯分类器。

以下是我的代码的一部分:

from sklearn.feature_extraction.text import CountVectorizer
matrix = CountVectorizer(ngram_range=(1,1))
X = matrix.fit_transform(data).toarray()
y =  [re.sub('[^A-Za-z]', ' ', y).strip(' ') for y in mobiles.iloc[:, 2]]

# split train and test data
from sklearn.model_selection import train_test_split
X_train, X_test, y_train, y_test = train_test_split(X, y)

# Naive Bayes 
from sklearn.naive_bayes import GaussianNB

classifier = GaussianNB()

classifier.fit(X_train, y_train)

# predict class
y_pred = classifier.predict(X_test)

res = pd.DataFrame({'y_test':y_test, 'y_pred':y_pred})
print(res)
# Confusion matrix
from sklearn.metrics import confusion_matrix, classification_report, accuracy_score, f1_score
cm = confusion_matrix(y_test, y_pred)
cr = classification_report(y_test, y_pred)

brands = list(set(y))

accuracy = accuracy_score(y_test, y_pred)
print("accuracy:", accuracy)
print("Confusion Matrix:")
import seaborn as sn
import pandas  as pd
import matplotlib.pyplot as plt

aylabels = brands #[str(i) for i in aylabels]
axlabels = brands #[str(i) for i in range(50)]

plt.figure(figsize=(10, 10))
sn.set(font_scale=1.4)  # for label size
sn.heatmap(cm, annot=True, annot_kws={"size": 12}, xticklabels=axlabels, yticklabels=aylabels)  # font size
plt.show()

以下是我在上面的代码中使用hitmap构建的混淆矩阵。虽然我设置aylabelsaxlabels同样的事情,但行和列在情节中是不同的。

enter image description here

我不知道发生了什么!

答案

你忘了在confusion_matrix中设置labels参数。

cm = confusion_matrix(y_test, y_pred, labels=brands)

以上是关于混淆矩阵与轴的不同标签的主要内容,如果未能解决你的问题,请参考以下文章

Scikit 学习如何打印混淆矩阵的标签?

不正确的混淆矩阵图

如何最小化平面物体的旋转检查,同时检查它们与轴的并行化

Python混淆矩阵可视化:plt.colorbar函数自定义颜色条的数值标签配置不同情况下颜色条的数值范围以及数据类型(整型浮点型)

如何使用 Tensorflow 创建预测标签和真实标签的混淆矩阵?

如何使用 Tensorflow 创建预测标签和真实标签的混淆矩阵?