首页 > 解决方案 > 如何将新向量添加到 Keras 嵌入矩阵

问题描述

背景

我在 Keras 中为分类数据列使用嵌入层。
我的理解是嵌入层只是一个由可训练向量组成的矩阵,每个向量都映射到一个索引。

我的问题

训练完成后,我想在嵌入矩阵中添加一个新的索引向量对。
(向量由我生成,现阶段不涉及任何训练。)

我该怎么做呢?
我也希望在预测中使用新添加的嵌入。

代码

keras.layers.Embedding(number_of_categories, embedding_size, input_length=1)

我特别卡住了,因为模型架构中编码了许多类别。有没有办法解决?

标签: machine-learningkerasdeep-learningcategorical-dataembedding

解决方案


推荐阅读