tf.contrib.learn.ModelFnOps

Ops returned from a model_fn.

THIS CLASS IS DEPRECATED. See contrib/learn/README.md for general migration instructions.

Args
mode One of ModeKeys. Specifies if this training, evaluation or prediction.
predictions Predictions Tensor or dict of Tensor.
loss Training loss Tensor.
train_op Op for the training step.
eval_metric_ops Dict of metric results keyed by name. The values of the dict are the results of calling a metric function, such as Tensor.
output_alternatives a dict of {submodel_name: (problem_type, {tensor_name: Tensor})}, where submodel_name is a submodel identifier that should be consistent across the pipeline (here likely taken from the name of each Head, for models that use them), problem_type is a ProblemType, tensor_name is a symbolic name for an output Tensor possibly but not necessarily taken from PredictionKey, and Tensor is the corresponding output Tensor itself.
training_chief_hooks A list of SessionRunHook objects that will be run on the chief worker during training.
training_hooks A list of SessionRunHook objects that will be run on all workers during training.
scaffold A tf.compat.v1.train.Scaffold object that can be used to set initialization, saver, and more to be used in training.
Raises
ValueError If validation fails.
Attributes
predictions
loss
train_op
eval_metric_ops
output_alternatives
training_chief_hooks
training_hooks
scaffold
mode

Methods

estimator_spec

View source

Creates an equivalent EstimatorSpec.

Args
default_serving_output_alternative_key Required for multiple heads. If you have multiple entries in output_alternatives dict (comparable to multiple heads), EstimatorSpec requires a default head that will be used if a Servo request does not explicitly mention which head to infer on. Pass the key of the output alternative here that you want to designate as default. A separate ExportOutpout for this default head will be added to the export_outputs dict with the special key saved_model.DEFAULT_SERVING_SIGNATURE_DEF_KEY, unless there is already an enry in output_alternatives with this special key.
Returns
Instance of EstimatorSpec that is equivalent to this ModelFnOps
Raises
ValueError If problem type is unknown.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/learn/ModelFnOps