pytorch深度学习求助帖

混淆矩阵只有第一行有数字输出

求助下大佬们哪里出了问题

# 可视化混淆矩阵
def plot_confusion_matrix(y_true, y_pred, classes):
    cm = confusion_matrix(y_true, y_pred)
    plt.figure(figsize=(8, 6))
    sns.heatmap(cm, fmt='d', cmap='Blues', xticklabels=classes, yticklabels=classes,annot=True, )
    plt.xlabel('Predicted labels')
    plt.ylabel('True labels')
    plt.title('Confusion Matrix')
    plt.show()

分类报告的显示是正常的 说明测试集应当是每个label都有覆盖的（但是不知道为什么第一个label只有199）