def METHOD_NAME(self, dataset):
    """Transform the target dataset for the model.
    # Arguments
        dataset: tf.data.Dataset. The target dataset for the model.
    # Returns
        An instance of tf.data.Dataset. The transformed dataset.
    """
    return self._transform_data(dataset, self.outputs)