在新模型中重用预训练的嵌入模型仅用于评估-keras

时间:2019-03-18 17:29:19

标签: python keras keras-layer

我正在寻找一种在新模型中重用经过训练的嵌入函数,但仅用于评估的方法。 例如,

def embedding(x):
     return Dense(2000)(x)

input = Input((4,200))
embedded = []
for index in range(4):
     embedded.append(embedding(Lambda(lambda x: x[:, lidx, :])(input)))
output = CustomLayer()(embedded)
model = Model(inputs=[input], output=[output]
model.compile()
model.fit()

我想用相同的受过训练的嵌入功能测试模型,但是现在输入具有6个元素而不是4个元素。

我如何重用嵌入函数和模型来做到这一点,而又不提取为K.function并遍历新的元素列表?

0 个答案:

没有答案
相关问题