在TensorFlow中,可以使用tf.keras.models.save_model和tf.keras.models.load_model函數來保存和加載模型。
tf.keras.models.save_model
tf.keras.models.load_model
保存模型:
model.save("my_model.h5")
加載模型:
model = tf.keras.models.load_model("my_model.h5")
這樣就可以將模型保存為一個.h5文件,并且在需要的時候加載模型。
億速云公眾號
手機網站二維碼
Copyright ? Yisu Cloud Ltd. All Rights Reserved. 2018 版權所有
廣州億速云計算有限公司粵ICP備17096448號-1 粵公網安備 44010402001142號增值電信業務經營許可證編號:B1-20181529