QBoard » Artificial Intelligence & ML » AI and ML - Conceptual » Which Loss function is correct for binary mapping?
User Dashboard

Which Loss function is correct for binary mapping?

Back To Topics
Tags : ai ml neural_nets
  • 106 5
    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?
      June 11, 2019 4:52 PM IST
    0
  • 90 6
    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.
      June 14, 2019 10:57 AM IST
    0
  • 129 8
    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.
      June 11, 2019 5:01 PM IST
    0