Define an Estimator Specification
estimator_spec(mode, predictions = NULL, loss = NULL, train_op = NULL, eval_metric_ops = NULL, training_hooks = NULL, evaluation_hooks = NULL, prediction_hooks = NULL, training_chief_hooks = NULL, ...)
A key that specifies whether we are performing
The prediction tensor(s).
The training loss tensor. Must be either scalar, or with shape
The training operation -- typically, a call to
A list of metrics to be computed as part of evaluation.
This should be a named list, mapping metric names (e.g.
(Available since TensorFlow v1.4) A list of session run hooks to run on all workers during training.
(Available since TensorFlow v1.4) A list of session run hooks to run during evaluation.
(Available since TensorFlow v1.7) A list of session run hooks to run during prediciton.
(Available since TensorFlow v1.4) A list of session run hooks to run on chief worker during training.
Other optional (named) arguments, to be passed to the