def build_decoder(self,input_shape):
data_dim = np.prod(input_shape)
return [
*[
Sequential([
Dense(self.parameters['layer'], activation=self.parameters['decoder_activation'], use_bias=False),
BN(),
Dropout(self.parameters['dropout']),])
for i in range(self.parameters['decoder_layers'])
],
Sequential([
Dense(data_dim, activation='sigmoid'),
Reshape(input_shape),]),]
评论列表
文章目录