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