二分类混淆矩阵

概念讲解

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

x是原本的标签，y是预测的标签

首字母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, local_labels = local_batch.