tf.estimator.Estimator

Estimator class to train and evaluate TensorFlow models.

Inherits From: Estimator

Used in the notebooks

Used in the guide Used in the tutorials

The Estimator object wraps a model which is specified by a model_fn, which, given inputs and a number of other parameters, returns the ops necessary to perform training, evaluation, or predictions.

All outputs (checkpoints, event files, etc.) are written to model_dir, or a subdirectory thereof. If model_dir is not set, a temporary directory is used.

The config argument can be passed