路缘石中的转移掩蔽
keras-trans-mask的Python项目详细描述
路缘石转移掩蔽
移除并还原不支持遮罩的层的遮罩。请注意,在大多数情况下,结果可能不正确。
安装
pip install keras-trans-mask
用法
Conv1D
不支持掩蔽。去掉遮罩,您将获得“几乎正确”的输出:
importkerasfromkeras_trans_maskimportRemoveMask,RestoreMaskinput_layer=keras.layers.Input(shape=(None,))embed_layer=keras.layers.Embedding(input_dim=10,output_dim=15,mask_zero=True,)(input_layer)removed_layer=RemoveMask()(embed_layer)# Remove mask from embeddingsconv_layer=keras.layers.Conv1D(filters=32,kernel_size=3,padding='same',)(removed_layer)restored_layer=RestoreMask()([conv_layer,embed_layer])# Restore mask from embeddingslstm_layer=keras.layers.LSTM(units=5)(restored_layer)dense_layer=keras.layers.Dense(units=2,activation='softmax')(lstm_layer)model=keras.models.Model(inputs=input_layer,outputs=dense_layer)model.compile(optimizer='adam',loss='sparse_categorical_crossentropy')model.summary()