text_model.py 文件源码

 复制代码def KeywordLayer(sequence_length, embed_input, embedding_dim, embed_matrix):
    model = Sequential()
    model.add(Embedding(embed_input, embedding_dim, input_length=sequence_length, weights=[embed_matrix]))
    model.add(GlobalMaxPooling1D())
    return model