简体   繁体   中英

Get predictions from Keras/Tensorflow once model is trained

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

在此处输入图像描述

在此处输入图像描述

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

You can use the .predict() function which you can pass new inputs into it and it give you a prediction. The docs for this function are here: keras

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM