Wie SigmoidCrossEntropyLoss in Caffe für binäre Klassenklassifikation verwenden?

Question

Wie SigmoidCrossEntropyLoss in Caffe für binäre Klassenproblem zu verwenden? Können Sie mir ein Prototyp-Beispiel geben?

Answer 1

Sie müssen Ihren label Eingang entweder Null oder eins pro Trainingsbeispiel eingeben. anders als das

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