Keras/Tensorflow: model.predict() returns a list. How do I match the output with my class names?
Model prediction output is a bunch of probabilities. In order to get category name you need use following snippet. It calculates the argmax of predicions and give it to CLASSES list:
print(CLASSES[np.argmax(predictions)])