tf.estimator.experimental.call_logit_fn

Calls logit_fn (experimental).

THIS FUNCTION IS EXPERIMENTAL. Keras layers/models are the recommended APIs for logit and model composition.

A utility function that calls the provided logit_fn with the relevant subset of provided arguments. Similar to tf.estimator._call_model_fn().

Args
logit_fn A logit_fn as defined above.
features The features dict.
mode TRAIN / EVAL / PREDICT ModeKeys.
params The hyperparameter dict.
config The configuration object.
Returns
A logit Tensor, the output of logit_fn.
Raises
ValueError if logit_fn does not return a Tensor or a dictionary mapping strings to Tensors.

© 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/r2.4/api_docs/python/tf/estimator/experimental/call_logit_fn