tensorflow - Predict if probability is higher than certain value

Question

I am using a MLP model for classification.

When I predict for new data, I want to keep only those predictions whose probability of prediction is larger than 0.5, and change all other predictions into class 0.

How can I do it in keras ?

I'm using using last layer as follows model.add(layers.Dense(7 , activation='softmax'))

Is it meaningful to get predictions with probability larger than 0.5 using the softmax?

newdata = (nsamples, nfeatures)
predictions = model.predict (newdata)
print (predictions.shape)
(500, 7)

question from:https://stackoverflow.com/questions/65882997/predict-if-probability-is-higher-than-certain-value

Answer 1

you can do something like:

preds=model.predict etc
index=np.argmax(preds)
probability= preds(index)
if probability >=.75:
    print (' class is ', index,' with high confidence')
elif probability >=.5:
    print (' class is ', index,' with medium confidence')
else:
    print (' class is ', index,' with low confidence')