pytorch画混淆矩阵

在PyTorch中，可以使用matplotlib库来绘制混淆矩阵。下面是一个示例代码，展示了如何使用PyTorch和matplotlib来画混淆矩阵：

import numpy as np
import matplotlib.pyplot as plt

# 定义混淆矩阵
def confusion_matrix(preds, labels, conf_matrix):
    for p, t in zip(preds, labels):
        conf_matrix\[p, t\] += 1
    return conf_matrix

# 绘制混淆矩阵
def plot_matrix(matrix, labels):
    plt.imshow(matrix, cmap=plt.cm.Blues)
    for x in range(len(labels)):
        for y in range(len(labels)):
            info = matrix\[y, x\]
            plt.text(x, y, info, verticalalignment='center', horizontalalignment='center')
    plt.xticks(range(len(labels)), labels, rotation=45)
    plt.yticks(range(len(labels)), labels)
    plt.xlabel('Predicted Label')
    plt.ylabel('True Label')
    plt.title('Confusion Matrix')
    plt.colorbar()
    plt.show()

# 示例代码
preds = \[0, 1, 2, 1, 0, 2, 2\]
labels = \[0, 1, 2, 1, 0, 1, 2\]
conf_matrix = np.zeros((3, 3))
conf_matrix = confusion_matrix(preds, labels, conf_matrix)
plot_matrix(conf_matrix, \['Label 0', 'Label 1', 'Label 2'\])

这段代码首先定义了一个混淆矩阵函数`confusion_matrix`，用于计算混淆矩阵。然后定义了一个绘制混淆矩阵的函数`plot_matrix`，该函数使用matplotlib库来绘制混淆矩阵。最后，通过调用这两个函数，可以计算准确率并绘制混淆矩阵。

希望这个示例代码能够帮助你理解如何使用PyTorch和matplotlib来画混淆矩阵。如果有任何问题，请随时提问。
#### 引用[.reference_title]
- *1* *2* [混淆矩阵的绘制](https://blog.csdn.net/qq_45470799/article/details/123737859)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control_2,239^v3^insert_chatgpt"}} ] [.reference_item]
- *3* [pytorch绘制混淆矩阵](https://blog.csdn.net/qq_18617009/article/details/103345308)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control_2,239^v3^insert_chatgpt"}} ] [.reference_item]
[ .reference_list ]