diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2016-09-15 16:40:05 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-09-15 17:46:33 -0700 |
commit | 0254040d28bf5cd4b77f195cfee5f15725634035 (patch) | |
tree | ff9f5fde31ba4f726de148a1f47d573053d5101e | |
parent | 0c7a0bbcafd5328cf977b41eb75613950167b02a (diff) |
Add back model_dir property to LinearClassifier that was lost in the class hierarchy refactoring.
Change: 133330855
-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 8ab43bcb1e..a61d87e1c0 100644 --- a/tensorflow/contrib/learn/python/learn/estimators/linear.py +++ b/tensorflow/contrib/learn/python/learn/estimators/linear.py @@ -590,6 +590,10 @@ class LinearClassifier(evaluable.Evaluable, trainable.Trainable): def config(self): return self._estimator.config + @property + def model_dir(self): + return self._model_dir + # TODO(zoy): Use model_fn similar to LinearClassifier. class LinearRegressor(dnn_linear_combined.DNNLinearCombinedRegressor): |