| View source on GitHub |
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. |
© 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