from tensorflow.contrib.learn.python.learn.estimators import head as head_lib

        config=config,
        params={
            "head":
                head_lib._regression_head(  # pylint: disable=protected-access
                    label_dimension=label_dimension,
                    weight_column_name=weight_column_name,
