def METHOD_NAME(head, feature_columns):
  return estimator.Estimator(
      model_fn=_base_model_fn,
      params={
          'model':
              composable_model.LinearComposableModel(
                  num_label_columns=head.logits_dimension),
          'feature_columns':
              feature_columns,
          'head':
              head
      })