2
$\begingroup$

I want to create a deep learning model (CNN) for binary classification, can I used the softmax function instead of the sigmoid function in binary classification?

Adding the classification layer to the model, will be like this

model.add(Dense(2, activation='softmax'))
$\endgroup$
4
$\begingroup$

For binary classification, it should give the same results, because softmax is a generalization of sigmoid for a larger number of classes.

$\endgroup$

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service, privacy policy and cookie policy

Not the answer you're looking for? Browse other questions tagged or ask your own question.