def METHOD_NAME(self, input_shape):
    layer_kwargs = dict(
        kernel_initializer=self.kernel_initializer,
        bias_initializer=self.bias_initializer,
        kernel_regularizer=self.kernel_regularizer,
        bias_regularizer=self.bias_regularizer,
        kernel_constraint=self.kernel_constraint,
        bias_constraint=self.bias_constraint,
    )
    self.mlp = Sequential(
        [
            Dense(channels, self.mlp_activation, **layer_kwargs)
            for channels in self.mlp_hidden
        ]
        + [Dense(self.k, "softmax", **layer_kwargs)]
    )
    super().METHOD_NAME(input_shape)