Get class labels from Keras functional model January 29, 2023 by Tarik y_prob = model.predict(x) y_classes = y_prob.argmax(axis=-1) As suggested here.