TensorFlow 2.0 Beta is available Learn more

tf.contrib.estimator.call_logit_fn

Calls logit_fn.

tf.contrib.estimator.call_logit_fn(
    logit_fn,
    features,
    mode,
    params,
    config
)

Defined in contrib/estimator/python/estimator/logit_fns.py.

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.