如何在caffe中使用SigmoidCrossEntropyLoss进行二进制类分类?

时间:2016-04-22 13:38:42

标签: deep-learning caffe matcaffe

如何在caffe中使用SigmoidCrossEntropyLoss来解决二进制类问题? 你能给我一个原型文例吗?

1 个答案:

答案 0 :(得分:2)

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

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