classifications = model.predict(test_images)
print(classifications[97])
>>> [6.1059264e-14 2.8641389e-06 8.3220975e-06 7.9902960e-04 1.3495819e-08 5.7034008e-14 1.5956934e-33 9.9918967e-01 6.7410788e-08 1.1945669e-08]
np.argmax(classifications[97])
>>> 7