在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文件,并且在需要的時候加載模型。