tf.estimator.experimental.call_logit_fn
Calls logit_fn (experimental).
tf.estimator.experimental.call_logit_fn(
logit_fn, features, mode, params, config
)
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. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/estimator/experimental/call_logit_fn