我试图了解如何应用掩蔽,并确保我的输出被掩蔽。我使用以下模型来屏蔽输入:
def tdcnn2d_bilstm_mask():
with tf.device('/gpu:2'):
inputlayer1 = Input(shape = input_shape1)
x = TimeDistributed(Conv2D(4, kernel_size=(3, 3), padding='same', activation='relu'))(inputlayer1)
x = TimeDistributed(MaxPooling2D(pool_size=(2,2)))(x)
x = TimeDistributed(Conv2D(8, kernel_size=(3, 3), padding='same', activation='relu'))(x)
x = TimeDistributed(MaxPooling2D(pool_size=(2,2)))(x)
x = TimeDistributed(Conv2D(8, kernel_size=(3, 3), padding='same', activation='relu'))(x)
x = TimeDistributed(MaxPooling2D(pool_size=(2,2)))(x)
x = TimeDistributed(Conv2D(8, kernel_size=(3, 3), padding='same', activation='relu'))(x)
x = TimeDistributed(MaxPooling2D(pool_size=(2,2)))(x)
x = TimeDistributed(Flatten())(x)
mask = Masking(mask_value=0.)(x)
x = Bidirectional(LSTM(64, dropout=0.5, return_sequences=True))(mask)
out1 = TimeDistributed(Dense(num_class,activation='softmax'))(x)
model = keras.Model(inputs = [inputlayer1], outputs = [out1])#, out2, out3])
opt = Adam(lr=1e-3, decay=1e-3 / 200)
model.compile(loss = 'categorical_crossentropy', optimizer=opt,metrics = ['accuracy'])
return model
我的问题是:
最后一个不相关的问题是:
最好的
埃姆雷
目前没有回答
相关问题 更多 >
编程相关推荐