def build_decoder(self,input_shape):
data_dim = np.prod(input_shape)
return [
*([Dropout(self.parameters['dropout'])] if self.parameters['dropout_z'] else []),
Dense(self.parameters['layer'], activation='relu', use_bias=False),
BN(),
Dropout(self.parameters['dropout']),
Dense(self.parameters['layer'], activation='relu', use_bias=False),
BN(),
Dropout(self.parameters['dropout']),
Dense(data_dim, activation='sigmoid'),
Reshape(input_shape),]
评论列表
文章目录