python - Get predictions from Keras/Tensorflow once model is trained

Question

I am working on a project involving neural machine translation (translating English to French).

I have worked through some examples online, and have now finished the model. Once a model is trained using Keras, how do I then get a prediction of a translation without training the entire model again, because with the large dataset I am using, each epoch takes some time and of course, I can't train the model every time I want a translation.

So what is the correct way of then generating predictions on new inputs without training the whole model again?

Thanks

question from:https://stackoverflow.com/questions/65923671/get-predictions-from-keras-tensorflow-once-model-is-trained

Answer 1

You need to save your model the model and its weights when the fit ends using :

keras.model.save(model_name)

At any time, you can load your trained model using

model = keras.load(model_name)

then perform predictions as

y_pred = model.predict(x_test)

Hope this will be helpful