首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >如何在keras程序中调用TimeDistributed包装器中的层

如何在keras程序中调用TimeDistributed包装器中的层
EN

Stack Overflow用户
提问于 2018-04-19 14:49:23
回答 2查看 631关注 0票数 0

我想在具有给定架构的训练有素的sentEncoder模型中的第二个TimeDistributed包装器中调用‘Keras’模型中的层,以创建另一个用于进行中间预测的Keras模型。从'docEncoder‘模型中调用层非常简单,但是如何在训练'docEncoder’之后访问sentEncoder中的层呢?

代码语言:javascript
复制
l2_reg = regularizers.l2(reg_param)
sentence_input = Input(shape=(MAX_SENT_LENGTH,), dtype='int32')
embedding_layer = get_embedding_layer(embedding_dim=embedding_dim, embedding_matrix=embedding_matrix,
                                      max_num_words=max_num_words, max_sent_length=max_sent_length,
                                      reg_param=reg_param)
embedded_sequences = embedding_layer(sentence_input)
l_lstm = Bidirectional(GRU(gru_units, return_sequences=True))(embedded_sequences)
l_dense = TimeDistributed(Dense(2*gru_units, use_bias=True,
                                bias_initializer='zero',
                                kernel_initializer='glorot_uniform',
                                activation='tanh'))(l_lstm)
l_att = AttLayer()(l_dense)
sentEncoder = Model(sentence_input, l_att)
print(sentEncoder.summary())

doc_input = Input(shape=(max_sents, max_sent_length), dtype='int32')
review_encoder = TimeDistributed(sentEncoder)(doc_input)
l_lstm_sent = Bidirectional(GRU(gru_units, return_sequences=True))(review_encoder)
l_dense_sent = TimeDistributed(Dense(2*gru_units, use_bias=True,
                                     bias_initializer='zero',
                                     kernel_initializer='glorot_uniform',
                                     activation='tanh'))(l_lstm_sent)
l_att_sent = AttLayer()(l_dense_sent)
preds = Dense(n_classes, activation='sigmoid', kernel_regularizer=l2_reg)(l_att_sent)
docEncoder = Model(doc_input, preds)
EN

Stack Overflow用户

发布于 2018-06-13 05:36:49

可以像这样从TimeDistributed包装器中的层提取权重:

代码语言:javascript
复制
sent_att_u_weights = model.layers[4].get_weights()[0]
票数 1
EN
查看全部 2 条回答
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/49914452

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档