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,