keras加载lstm+crf模型出错怎么办-创新互联
小编这次要给大家分享的是keras加载lstm+crf模型出错怎么办,文章内容丰富,感兴趣的小伙伴可以来了解一下,希望大家阅读完这篇文章之后能够有所收获。
成都创新互联公司专注于翁源企业网站建设,响应式网站设计,成都商城网站开发。翁源网站建设公司,为翁源等地区提供建站服务。全流程定制设计,专业设计,全程项目跟踪,成都创新互联公司专业和态度为您提供的服务错误展示
new_model = load_model(“model.h6”)
报错:
1、keras load_model valueError: Unknown Layer :CRF
2、keras load_model valueError: Unknown loss function:crf_loss
错误修改
1、load_model修改源码:custom_objects = None 改为 def load_model(filepath, custom_objects, compile=True):
2、new_model = load_model(“model.h6”,custom_objects={‘CRF': CRF,‘crf_loss': crf_loss,‘crf_viterbi_accuracy': crf_viterbi_accuracy}
以上修改后,即可运行。
Code Example:
# coding: utf-8 from keras.models import Sequential from keras.layers import Embedding from keras.layers import LSTM from keras.layers import Bidirectional from keras.layers import Dense from keras.layers import TimeDistributed from keras.layers import Dropout from keras_contrib.layers.crf import CRF from keras_contrib.utils import save_load_utils VOCAB_SIZE = 2500 EMBEDDING_OUT_DIM = 128 TIME_STAMPS = 100 HIDDEN_UNITS = 200 DROPOUT_RATE = 0.3 NUM_CLASS = 5 def build_embedding_bilstm2_crf_model(): """ 带embedding的双向LSTM + crf """ model = Sequential() model.add(Embedding(VOCAB_SIZE, output_dim=EMBEDDING_OUT_DIM, input_length=TIME_STAMPS)) model.add(Bidirectional(LSTM(HIDDEN_UNITS, return_sequences=True))) model.add(Dropout(DROPOUT_RATE)) model.add(Bidirectional(LSTM(HIDDEN_UNITS, return_sequences=True))) model.add(Dropout(DROPOUT_RATE)) model.add(TimeDistributed(Dense(NUM_CLASS))) crf_layer = CRF(NUM_CLASS) model.add(crf_layer) model.compile('rmsprop', loss=crf_layer.loss_function, metrics=[crf_layer.accuracy]) return model def save_embedding_bilstm2_crf_model(model, filename): save_load_utils.save_all_weights(model,filename) def load_embedding_bilstm2_crf_model(filename): model = build_embedding_bilstm2_crf_model() save_load_utils.load_all_weights(model, filename) return model if __name__ == '__main__': model = build_embedding_bilstm2_crf_model()
另外有需要云服务器可以了解下创新互联scvps.cn,海内外云服务器15元起步,三天无理由+7*72小时售后在线,公司持有idc许可证,提供“云服务器、裸金属服务器、高防服务器、香港服务器、美国服务器、虚拟主机、免备案服务器”等云主机租用服务以及企业上云的综合解决方案,具有“安全稳定、简单易用、服务可用性高、性价比高”等特点与优势,专为企业上云打造定制,能够满足用户丰富、多元化的应用场景需求。
网站名称:keras加载lstm+crf模型出错怎么办-创新互联
网站链接:http://azwzsj.com/article/jspeh.html