python 混淆矩阵热力图_混淆矩阵

要绘制 Python 中的混淆矩阵热力图，我们可以使用 matplotlib 库和 seaborn 库。以下是一个示例代码：

首先，请确保你已经安装了 matplotlib 和 seaborn 库。

import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns

# 假设我们有一个混淆矩阵
confusion_matrix = np.array([[40, 5],
                             [10, 25]])

# 绘制热力图
plt.figure(figsize=(8, 6))
sns.heatmap(confusion_matrix, annot=True, fmt='d', cmap='Blues')
plt.xlabel('Predicted')
plt.ylabel('Actual')
plt.title('Confusion Matrix Heatmap')
plt.show()
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.
  • 6.
  • 7.
  • 8.
  • 9.
  • 10.
  • 11.
  • 12.
  • 13.
  • 14.
  • 15.

在上述代码中，我们首先定义了一个混淆矩阵 confusion_matrix，然后使用 seaborn 的 heatmap 函数绘制热力图。annot=True 表示在每个单元格中显示数值，fmt='d' 表示以整数形式显示数值，cmap='Blues' 表示使用蓝色的颜色映射。

你可以根据自己的实际混淆矩阵数据进行修改和调整。希望这个示例对你有帮助！如果还有其他问题，随时可以问我。