import numpy as np

​

prediction = model.prediction(test_data) 

# prediction will contain [[0.6, 0.05, 0.05, 0.05, 0.05, 0.05, 0.05, 0.05, 0.05]]

​

prediction = np.argmax(prediction[0])

# Now predition hold the index of the (0.6) i.e max probability value