如何在咖啡馆中使用 Sigmoid 交叉熵损失进行二进制 class class 化？

Question

如何在 cafe 中使用 Sigmoid Cross Entropy Loss 解决二进制 class 问题？ 你能给我一个 prototxt 的例子吗？

Answer 1

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

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