Keras嵌入层和CNN:如何从模型中获取训练后的向量

时间:2018-07-08 21:00:05

标签: tensorflow plot keras embedding

我使用了包含嵌入层的神经网络。

我将数据集(由包含句子的多个文件组成)用Tokenizer转换为向量,并将它们作为训练输入馈送到网络。嵌入层之后,我得到了一个CNN。我怎么知道将训练后的输入向量取回作图?

1 个答案:

答案 0 :(得分:2)

很简单,尝试:

for layer in my_model.layers:
    weights = layer.get_weights()
    configs = layer.get_config()

您必须在配置中进行一些解析才能检索嵌入层。您拥有可用于识别和检索图层的“名称”