keras中保存模型

1.首先要安装h5py库

sudo pip install h5py

2.在训练完后保存模型,若保存为单一文件使用model.save()，里面包含了模型的结构和权重。

model = Sequential() 

model.add(LSTM(...)) 


model.compile(...) 


model.fit(...) 

# save model 

model.save('lstm_model.h5' )

可以在其他文件中加载出来预测。注意在predict函数中，加入verbose=0,否则会报错。

 from keras.models import load_model 

# load model from single file 

model = load_model('lstm_model.h5') 

# make predictions 

yhat = model.predict(X, verbose=0) #在这里输入你的测试集x

print(yhat)