Skip to main content
added 224 characters in body
Source Link
Gerry P
  • 191
  • 4

Is this a multi-label problem or a pure classification problem? If it is just classification change the activation function in the final layer to softmax. Then train andWhen you do predictions. Select select the prediction that hasoutput with the highest probability as the class. Alternatively use model.evaluate(.....). versus model.predict Documentation is here,

change the activation function in the final layer to softmax. Then train and do predictions. Select the prediction that has the highest probability.

Is this a multi-label problem or a pure classification problem? If it is just classification change the activation function in the final layer to softmax. When you do predictions select the output with the highest probability as the class. Alternatively use model.evaluate(.....). versus model.predict Documentation is here,

Source Link
Gerry P
  • 191
  • 4

change the activation function in the final layer to softmax. Then train and do predictions. Select the prediction that has the highest probability.