首页 > 解决方案 > 在 keras 中加载模型

问题描述

在 keras API 中将模型(先前处理过的)加载到 LSTM 网络中时,它是否会在网络中(显式)包含训练数据?

如果没有,这将如何 (加载模型)帮助预测时间序列中的模式。

标签: kerasdeep-learninglstm

解决方案


from keras.models import Sequential
from keras_contrib.losses import import crf_loss
from keras_contrib.metrics import crf_viterbi_accuracy

# To save model
model.save('my_model_01.hdf5')

# To load the model
custom_objects={'CRF'                 : CRF,
              'crf_loss'            : crf_loss,
              'crf_viterbi_accuracy': crf_viterbi_accuracy}

# To load a persisted model that uses the CRF layer 
model1 = load_model("/home/abc/my_model_01.hdf5", custom_objects = custom_objects)

推荐阅读