본문 바로가기

IT

softmax vs sigmoid

 

binary class 인데 one hot encoding을 했다?? -> softmax 써야함.

binary class 인데 label이 scalar다?(binary encoding) -> sigmoid

 

multi calss는 그냥 softmax 쓰자.

 

  • for binary_crossentropy: sigmoid activation, scalar target
  • for categorical_crossentropy: softmax activation, one-hot encoded target