diff options
Diffstat (limited to 'tensorflow/contrib/learn/python/learn/estimators/linear.py')
-rw-r--r-- | tensorflow/contrib/learn/python/learn/estimators/linear.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/contrib/learn/python/learn/estimators/linear.py b/tensorflow/contrib/learn/python/learn/estimators/linear.py index 45e430717f..c4a257b8d4 100644 --- a/tensorflow/contrib/learn/python/learn/estimators/linear.py +++ b/tensorflow/contrib/learn/python/learn/estimators/linear.py @@ -660,9 +660,10 @@ class LinearRegressor(evaluable.Evaluable, trainable.Trainable): """ self._feature_columns = feature_columns assert self._feature_columns - self._optimizer = _get_default_optimizer(feature_columns) if optimizer: self._optimizer = _get_optimizer(optimizer) + else: + self._optimizer = _get_default_optimizer(feature_columns) chief_hook = None if (isinstance(optimizer, sdca_optimizer.SDCAOptimizer) and |