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,以生成您自己的混淆矩阵可视化图表。