ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more


View source on GitHub

Calls logit_fn.

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

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.

A logit Tensor, the output of logit_fn.

ValueError if logit_fn does not return a Tensor or a dictionary mapping strings to Tensors.