我正在寻找一种在新模型中重用经过训练的嵌入函数,但仅用于评估的方法。 例如,
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并遍历新的元素列表?