keras - 如何将掩码值输入到 Convolution1D 层

Question

我需要将可变长度序列输入到我的模型中。

我的型号是 Embedding + LSTM + Conv1d + Maxpooling + softmax .

当我设置 mask_zero = True在 Embedding , 我在 Conv1d 编译失败.

如何在 Conv1d 中输入掩码值还是有其他解决方案？

Answer 1

Masking layer 期望每个下游层都支持掩码，这不是 Conv1D 的情况层。幸运的是，还有另一种应用屏蔽的方法，使用 Functional API:

inputs = Input(...)
mask = Masking().compute_mask(inputs) # <= Compute the mask
embed = Embedding(...)(inputs)
lstm = LSTM(...)(embed, mask=mask) # <= Apply the mask
conv = Conv1D(...)(lstm)
...
model = Model(inputs=[inputs], outputs=[...])