tf.estimator.experimental.call_logit_fn

Calls logit_fn (experimental).

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.estimator.experimental.call_logit_fn

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.