diff options
| author |  Phil Stahlfeld <philstahlfeld@google.com> | 2016-08-30 14:47:18 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2016-08-30 16:03:02 -0700 |
| commit | f61f2e570045de904f7a84f17fb68c9b00a4cd73 (patch) | |
| tree | ab7654d5cfa4e57e2f35ee5537820b7e7ccc77d8 | |
| parent | 197ab588b7f5faaeeccdac809d13e72a225e32d9 (diff) | |
Proxy LinearClassifier.config to internal estimator's config.
This allows LinearClassifier to continue functioning with learn_runner.
Change: 131765191
| -rw-r--r-- | tensorflow/contrib/learn/python/learn/estimators/linear.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/contrib/learn/python/learn/estimators/linear.py b/tensorflow/contrib/learn/python/learn/estimators/linear.py index 6450f675a3..74711e98b2 100644 --- a/tensorflow/contrib/learn/python/learn/estimators/linear.py +++ b/tensorflow/contrib/learn/python/learn/estimators/linear.py @@ -503,6 +503,10 @@ class LinearClassifier(evaluable.Evaluable, trainable.Trainable): return checkpoints.load_variable(self._model_dir, name="linear/bias_weight") + @property + def config(self): + return self._estimator.config + class LinearRegressor(dnn_linear_combined.DNNLinearCombinedRegressor): """Linear regressor model. |