It looks like you are using Keras or did you implement the same model in PyTorch?
If so, could you post the model architecture or the shape of your last linear layer?
I assume you’ve used a single output with nn.BCELoss/nn.BCEWithLogitsLoss or two output units with nn.CrossEntropyLoss/nn.NLLLoss.
In the first case, you can just apply a threshold (e.g. 0.5) or fine tune it using the ROC, which you would apply torch.argmax(output, 1) in the latter case to get the prediction.