def apply_final_layers(self, inputs):
    x = inputs
    if self.with_discriminator:
        if (self.with_discriminator is True):
            final_activation = 'sigmoid'
        else:
            final_activation = self.with_discriminator
        x = self.apply(inputs=x, layer=Dense, units=self.hidden_size, activation=self.hidden_act, kernel_initializer=self.initializer, name='Discriminator-Dense')
        x = self.apply(inputs=x, layer=Dense, units=1, activation=final_activation, kernel_initializer=self.initializer, name='Discriminator-Prediction')
    return x
def apply_final_layers(self, inputs):
    x = inputs
    if self.with_discriminator:
        if (self.with_discriminator is True):
            final_activation = 'sigmoid'
        else:
            final_activation = self.with_discriminator
        x = self.apply(inputs=x, layer=Dense, units=self.hidden_size, activation=self.hidden_act, kernel_initializer=self.initializer, name='Discriminator-Dense')
        x = self.apply(inputs=x, layer=Dense, units=1, activation=final_activation, kernel_initializer=self.initializer, name='Discriminator-Prediction')
    return x