from tensorflow.contrib.learn.python.learn.estimators import composable_model

    num_ps_replicas = config.num_ps_replicas if config else 0

    self._linear_model = composable_model.LinearComposableModel(
        num_label_columns=target_column.num_label_columns,
