spaCy

spaCy怎么可视化混淆矩阵

小亿
84
2024-05-11 17:49:55
栏目: 编程语言

spaCy库本身并不提供混淆矩阵的可视化功能,但可以借助其他库来实现。以下是一个示例代码,使用matplotlib库来可视化混淆矩阵:

from sklearn.metrics import confusion_matrix
import matplotlib.pyplot as plt

# 生成混淆矩阵
true_labels = [0, 1, 0, 1, 0, 1, 0, 1, 0, 1]
predicted_labels = [0, 1, 1, 1, 0, 1, 0, 0, 0, 1]

cm = confusion_matrix(true_labels, predicted_labels)

# 可视化混淆矩阵
plt.imshow(cm, interpolation='nearest', cmap=plt.cm.Blues)
plt.title("Confusion Matrix")
plt.colorbar()

tick_marks = [0, 1]
plt.xticks(tick_marks, tick_marks)
plt.yticks(tick_marks, tick_marks)

plt.ylabel('True label')
plt.xlabel('Predicted label')

plt.show()

在这个示例中,我们首先生成了一个混淆矩阵,然后使用matplotlib库来可视化混淆矩阵。您可以根据自己的数据和模型预测结果来替换示例中的true_labels和predicted_labels,以生成您自己的混淆矩阵可视化图表。

0
看了该问题的人还看了