TensorFlow 2.0 RC is available Learn more

tf.estimator.EstimatorSpec

TensorFlow 2.0 version View source on GitHub

Class EstimatorSpec

Ops and objects returned from a model_fn and passed to an Estimator.

Aliases:

  • Class tf.compat.v1.estimator.EstimatorSpec
  • Class tf.compat.v2.estimator.EstimatorSpec

EstimatorSpec fully defines the model to be run by an Estimator.

Properties

mode

predictions

loss

train_op

eval_metric_ops

export_outputs

training_chief_hooks

training_hooks

scaffold

evaluation_hooks

prediction_hooks