在解决二元分类问题时,我认为caffe有两种可能的方法
第一个是使用"SigmoidCrossEntropyLossLayer"
和一个输出单位
另一个是使用"SoftmaxWithLossLayer"
和两个输出单元。
我的问题是这两种方法之间的区别是什么?
我应该使用哪一个?
非常感谢你!
答案 0 :(得分:1)
如果你玩数学,你可以重复"对于类1 "Sigmoid"
层到0.5*x_i
的预测类概率,对于类0,-0.5*x_i
的预测类概率,单"SoftmaxWithLoss"
层的预计类概率为"SigmoindWithCrossEntropy"
预测x_i
。
所以我相信可以说这两种方法可以被认为是预测二进制输出的等效方法。