溫馨提示×

spaCy怎么可視化混淆矩陣

小億
93
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
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女