Using a python trained neural network

43 次查看(过去 30 天)
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?

采纳的回答

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

更多回答(0 个)

类别

Help CenterFile Exchange 中查找有关 Deep Learning Toolbox 的更多信息

产品


版本

R2019a

Community Treasure Hunt

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

Start Hunting!

Translated by