How to predict single image using the save model #1469
-
I have trained cats and dogs for using ImageClassifier and save the model with model.save("model_autokeras.h5"), now when I use the model for prediction I only getting the RESULT below, I am expecting 2 classes (cats and dogs), I want to know the class predicted and the proba for each. Can someone point me to some documentation please.
RESULT = [[0.9998584]] |
Beta Was this translation helpful? Give feedback.
Replies: 1 comment
-
For 2-class classification we are using a single output as the probability for the first class "1-p" would be the probability for the other class. |
Beta Was this translation helpful? Give feedback.
For 2-class classification we are using a single output as the probability for the first class "1-p" would be the probability for the other class.
In your case, the probability would be 0.9998584 and (1-0.9998584) for cats and dogs.