Using a python trained neural network

28 views (last 30 days)
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?

Accepted Answer

David Willingham
David Willingham on 25 Oct 2022
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")

More Answers (0)

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!