使用sklearn库里的模型
使用以下代码保存和读取都出现了错误。
joblib.dump(clf,'filename.pkl')
clf=joblib.load('filename.pkl')
使用Keras提供的一个API解决了
该模型以HDF5文件格式保存,所以先要安装h5py Python库,安装方法如下:
在cmd中直接pip install h5py
model = Sequential()
model.add(LSTM(...))
model.compile(...)
model.fit(...)
#保存模型
model.save('lstm_model.h5')
#读取模型
from keras.models import load_model
model = load_model("lstm.h5")
#进行预测
model.predict(x)