二分类混淆矩阵

confusion matrix

    • 概念讲解
    • 代码实现
    • 绘制热力图

概念讲解

x = local_labels[idx]
y = predicted[idx]

x是原本的标签,y是预测的标签
二分类混淆矩阵_第1张图片
首字母T,F,表示模型是否预测正确(T表示正确)
次字母N,P,表示预测成了阴性还是阳性(一般认为1是阳性)

代码实现

test_dataset = TensorDataset(Test_data, Test_label)
test_cm_gen = DataLoader(test_dataset, batch_size=Batch_size, shuffle=False)

cm_test = np.array([[0, 0 ], [0, 0 ]])
print(type(cm_test))
print(cm_test.shape)

with torch.set_grad_enabled(False):
  for local_batch, local_labels in test_cm_gen:
    # Transfer to GPU
    local_batch<

你可能感兴趣的:(深度学习,深度学习,机器学习,线性代数)