diff options
Diffstat (limited to 'tensorflow/contrib/learn/python/learn/estimators/logistic_regressor.py')
| -rw-r--r-- | tensorflow/contrib/learn/python/learn/estimators/logistic_regressor.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/contrib/learn/python/learn/estimators/logistic_regressor.py b/tensorflow/contrib/learn/python/learn/estimators/logistic_regressor.py index d4e0bb6283..cafdb980c5 100644 --- a/tensorflow/contrib/learn/python/learn/estimators/logistic_regressor.py +++ b/tensorflow/contrib/learn/python/learn/estimators/logistic_regressor.py @@ -57,7 +57,9 @@ class LogisticRegressor(estimator.Estimator): expects the returned predictions to be probabilities in [0.0, 1.0]. thresholds: List of floating point thresholds to use for accuracy, precision, and recall metrics. If None, defaults to [0.5]. - model_dir: Directory to save model parameters, graphs, etc. + model_dir: Directory to save model parameters, graphs, etc. This can also + be used to load checkpoints from the directory into a estimator to continue + training a previously saved model. config: A RunConfig configuration object. """ if thresholds is None: |