Как сохранить модель Terrasorflow Keras? - PullRequest
1 голос
/ 02 марта 2020

Я пытаюсь сохранить модель tenasflow keras с этой сводкой:

Model: "sequential_2" etc.

с помощью следующей команды:

model.save('my_model.h5')

Я получил следующую ошибку (в Jupyter Notebook):

ImportError: `save_model` requires h5py.

Итак, я установил h5py, используя conda install h5py После установки я получил версию h5py в блокноте Jupyter (там же, где я пытаюсь сохранить модель):

h5py.__version__
'2.8.0'

Тем не менее, я получил ту же ошибку. Хотя я импортировал h5py вручную.

import h5py
import tensorflow as tf
from tensorflow import keras
from tensorflow.keras import layers
filename = 'model.h5' 
model.save(filename)

---------------------------------------------------------------------------
**ImportError**                               Traceback (most recent call last)
<ipython-input-54-9160eee81fe6> in <module>
      5 from tensorflow.keras import layers
      6 filename = 'model.h5'
----> 7 model.save(filename)

~/anaconda3/lib/python3.7/site-packages/tensorflow_core/python/keras/engine/network.py in save(self, filepath, overwrite, include_optimizer, save_format, signatures)

~/anaconda3/lib/python3.7/site-packages/tensorflow_core/python/keras/saving/save.py in save_model(model, filepath, overwrite, include_optimizer, save_format, signatures)

~/anaconda3/lib/python3.7/site-packages/tensorflow_core/python/keras/saving/hdf5_format.py in save_model_to_hdf5(model, filepath, overwrite, include_optimizer)

**ImportError:** save_model requires h5py.

1 Ответ

0 голосов
/ 23 марта 2020

Вы пробовали напрямую установить h5py ?
Если у вас есть существующая установка Python (например, загрузка python .org или та, которая поставляется с вашей ОС), то на Windows, macOS / OSX и Linux на компьютерах Intel, предварительно установленные колеса h5py можно установить через pip из PyPI:

$ pip install h5py

In Windows:
Если это не удастся, вы можете попробовать установить cython
pip install h5py
pip install cython

В Linux / Ubuntu :
Если это не удастся, вам также может понадобиться libhdf5, установить libhdf5 и затем установить h5py
1. sudo pip install cython
2. sudo apt-get install libhdf5-dev
3. sudo pip install h5py

Примечание: Если вышеуказанная установка не сработала, попробуйте удалить h5py, а затем переустановить его и перезапустить ноутбук Jupyter или Anaconda.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...