# Trying to make predictions on a single image
predictions = model.predict(test_images)
checkPrediction(0)

def checkPrediction(index):
    probaArray = predictions[index]
    print("Model guessed: " + np.argmax(probaArray)             # display guess of the model
          + " | Real label is: " + test_labels[index]           # display the real classification
          + " | Whole probability array is: " + probaArray)     # display probability array