训练过的模型无法保存tensorflow keras

Question

我无法在tensorflow中保存经过训练的模型

我有适合的数据

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'

我不知道该怎么做才能保存此模型，请帮我将其保存在yaml文件中

ps：-我正在研究kaggle

Answer 1

在TensorFlow中保存模型时。您可以在模型的自身上调用save方法，而不是从history返回的fit or fit_generator对象上调用方法。

请使用此

  model.save('model.h5')