我需要将图像分类为癌性或非癌性。
为此，我构建了一个经典的 CNN，但我在使用两列向量标记我的数据集之间犹豫不决，如下所示：
cancerous: [0, 1]
not cancerous: [1, 0]
并使用具有 2 个输出神经元的 softmax 激活函数。
model.add(Dense(2, activation='softmax'))
或者
cancerous: [1]
not cancerous: [0]
并使用带有一个输出神经元的 sigmoid 激活函数。
model.add(Dense(1, activation='sigmoid'))
鉴于我需要使用患癌症的概率作为患者的最终指标并绘制 ROC 曲线，哪种模型更好？