Обученная модель не может быть сохранена tenorflow keras - PullRequest
0 голосов
/ 02 августа 2020

Я не могу сохранить обученную модель в тензорном потоке

Я подогнал данные

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

1 Ответ

0 голосов
/ 02 августа 2020

при сохранении модели в TensorFlow. Вы вызываете метод save в модели, а не в объекте history, возвращенном из fit or fit_generator.

Пожалуйста, используйте этот

  model.save('model.h5')
...