tf.contrib.learn.ModelFnOps

Ops returned from a model_fn.

tf.contrib.learn.ModelFnOps(
    mode, predictions=None, loss=None, train_op=None, eval_metric_ops=None,
    output_alternatives=None, training_chief_hooks=None, training_hooks=None,
    scaffold=None
)

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

estimator_spec(
    default_serving_output_alternative_key=None
)

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.