tensorflow中save與load模型

# 保存權重,該方式僅僅保存權重張量
model.save_weights('./data/tf_model_weights.ckpt',save_format = "tf")
# 保存模型結構與模型參數到文件,該方式保存的模型具有跨平臺性便於部署
model.save('./data/tf_model_savedmodel', save_format="tf")
model_loaded = tf.keras.models.load_model('./data/tf_model_savedmodel')
model_loaded.evaluate(x_test,y_test)
##輸出的是loss和AUC

參考源:https://github.com/lyhue1991/eat_tensorflow2_in_30_days

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章