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$
  • $\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$ – Dmytro Prylipko Jan 11 at 20:41
0
$\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 acknowledge that you have read our updated terms of service, privacy policy and cookie policy, and that your continued use of the website is subject to these policies.

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