2017-06-22 9 views
0

私はh5py形式で訓練したニューラルネットワークを保存できません。次のエラーが表示されます。ケラスを使って訓練されたニューラルネットワークを保存できません

ImportError        Traceback (most recent call last) 
<ipython-input-64-0185b568b480> in <module>() 
     1 from keras.models import load_model 
----> 2 model.save("MNISTclassifier.h5") 

/usr/local/lib/python3.5/dist-packages/keras/engine/topology.py in save(self, filepath, overwrite, include_optimizer) 
    2470   """ 
    2471   from ..models import save_model 
-> 2472   save_model(self, filepath, overwrite, include_optimizer) 
    2473 
    2474  def save_weights(self, filepath, overwrite=True): 

/usr/local/lib/python3.5/dist-packages/keras/models.py in save_model(model, filepath, overwrite, include_optimizer) 
    53 
    54  if h5py is None: 
---> 55   raise ImportError('`save_model` requires h5py.') 
    56 
    57  def get_json_type(obj): 

ImportError: `save_model` requires h5py. 

pipを使用してh5pyをインストールしてインポートしましたが、ここ ソースコード: https://github.com/tanmay-edgelord/HandwrittenDigitRecognition/blob/master/MNIST%20.ipynb

+1

は修正は[こちら](https://github.com/fchollet/keras/issues/3426)はあなたの問題を解決する提案しますか? – dhinckley

+0

PCを再起動すると私の問題が解決しました。私は自分のコードを変更しなかった。私がコメントしなかったように。 –

答えて

1

あなたが含まれています単一HDF5ファイルにKerasモデルを保存するためにmodel.save(filepath)を使用することができます。

モデルを再作成することができ
  1. モデルのアーキテクチャ、
  2. モデルの重み
  3. トレーニング構成(損失、オプティマイザ)
  4. オフラインで正確にトレーニングを再開できるオプティマイザの状態。

keras.models.load_model(filepath)を使用してモデルを再インスタンス化できます。 load_modelは、モデルが最初にコンパイルされていない限り、保存されたトレーニング構成を使用してモデルをコンパイルすることもあります。

例:

from keras.models import load_model 
model.save('my_model.h5') # creates a HDF5 file 'my_model.h5' 
+0

私はそれを理解したが、とにかく感謝する。 –

+0

いいですね。マシンラーニングでうまくやってください。 –

関連する問題