微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

在 Keras 中向 BiLSTM 模型添加 CRF 层2021 年 1 月

如何解决在 Keras 中向 BiLSTM 模型添加 CRF 层2021 年 1 月

我已经通过互联网搜索解决这个问题,但似乎没有人有解决方案。我想用 CRF 层改进我的 BiLSTM 模型,显然 Keras 不再支持它。我怎样才能添加这个?我不想为它更改我的整个 Keras 和 Tensorflow 版本。请帮帮我 :) 我有最新的 Keras 和 Tensorflow 版本。

    #vocab_size=4840,embedding is glove6B,max_seq_length=100
    model =  Sequential()
    model.add(Embedding(vocab_size,300,weights=[embedding_vectors],input_length=max_seq_length,trainable= False))
    model.add(Bidirectional(LSTM(units=50,dropout=0.5,recurrent_dropout=0.5,return_sequences=True)))
    model.add(Dense(512,activation='relu'))
    model.add(Dropout(0.4))
    model.add(Dense(32,activation='relu'))
    model.add(Dense(n_tags,activation='softmax'))
    

    model.compile(loss="categorical_crossentropy",optimizer="rmsprop",metrics= ["categorical_accuracy"])
    model.summary()

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。