deep-learning - 如何在 caffe 中使用 SigmoidCrossEntropyLoss 进行二元类分类？

Question

如何在 caffe 中使用 SigmoidCrossEntropyLoss 解决二元类问题？你能给我一个prototxt的例子吗？

Answer 1

您需要label每个训练样本的输入为零或一。除此之外

 layer {
   name: "loss"
   type: "SigmoidCrossEntropyLoss"
   bottom: "predict1d"
   bottom: "label"
   top: "loss"
 }