訓練后的 model 無法保存 tensorflow keras

Question

我無法在 tensorflow 中保存經過訓練的 model

我已經擬合了數據

results = model.fit_generator(train_image_gen,epochs=20,validation_data=test_image_gen,callbacks=early_stop)

並進口

from tensorflow.keras.models import load_model

並寫了這樣的代碼

save1= results.history
save1.save('model.h5')

但它會引發這樣的錯誤

AttributeError                            Traceback (most recent call last)
<ipython-input-47-13cfa06fe0cd> in <module>
----> 1 save1.save('model.h5')

AttributeError: 'dict' object has no attribute 'save'

我不知道該怎么做才能保存這個 model 請也幫助我如何將它保存在 yaml 文件中

ps:-我正在研究 kaggle

Answer 1

在 TensorFlow 中保存TensorFlow時。 您在 model 上調用save方法，它本身不在從fit or fit_generator返回的history object 上。

請用這個

  model.save('model.h5')