Using a python trained neural network

Can you please tell me how I can save a neural network after training,in Jupiter notebook and retrieve it in MATLAB for testing on the TEST dataset.

Should I store it without .h5 or with h5 extension? A new folder "MyModel" is created, when I run generator.save("MyModel") and there is a file inside it with name "saved_model.pb"? How it is loaded and tested in MATLAB 2019a version?

[Sania Gul]

Wow. That's great. Tnk u sooooooooo much.

[Sania Gul]

MATLAB is the BEST

Answer 1

[David Willingham]

Hi Sania,

I would recommend saving the model in the TensorFlow saved model format and then import it via the TensorFlow converter using the function importTensorFlowNetwork.

To save a model in the TensorFlow saved model format use the following python code:

model.save("MyModel")