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.
|