spaCy怎么可視化混淆矩陣

小億
84
2024-05-11 17:49:55
欄目: 編程語言

spaCy庫本身并不提供混淆矩陣的可視化功能，但可以借助其他庫來實(shí)現(xiàn)。以下是一個(gè)示例代碼，使用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()

在這個(gè)示例中，我們首先生成了一個(gè)混淆矩陣，然后使用matplotlib庫來可視化混淆矩陣。您可以根據(jù)自己的數(shù)據(jù)和模型預(yù)測結(jié)果來替換示例中的true_labels和predicted_labels，以生成您自己的混淆矩陣可視化圖表。

0 踩