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

    self._hidden_units = hidden_units
    self._feature_columns = tuple(feature_columns or [])
    self._enable_centered_bias = enable_centered_bias
    self._estimator = estimator.Estimator(
        model_fn=_dnn_model_fn,
        model_dir=model_dir,
        config=config,
        params={
            "head":
                head_lib._multi_class_head(  # pylint: disable=protected-access
                    n_classes,
                    weight_column_name=weight_column_name,
                    enable_centered_bias=enable_centered_bias),
            "hidden_units":
                hidden_units,
            "feature_columns":
                self._feature_columns,
            "optimizer":
