1
$\begingroup$

I have built a 3 layer neural network to perform a binary mapping (2016 inputs, 288 outputs.) I am getting decent results with mean square error and stochastic gradient decent. My question is: Is there a more appropriate loss function for regression when the output is binary?

$\endgroup$
2
  • $\begingroup$ Hmm, for binary output it is more convenient to have sigmoid (softmax is not appropriate in this case, I guess) output and cross-entropy loss function. $\endgroup$ Commented Jan 11, 2019 at 20:41
  • $\begingroup$ What is meant by binary? Are you saying two outputs always or what? Share the output format as well. $\endgroup$ Commented Feb 6, 2020 at 10:36

1 Answer 1

2
$\begingroup$

Yes, use binary cross-entropy loss. In case you are using Keras, this has been already implemented as a standard loss function for binary outputs.

$\endgroup$

Your Answer

By clicking “Post Your Answer”, you agree to our terms of service and acknowledge you have read our privacy policy.

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