tf.estimator.experimental

TensorFlow 1 version

View source on GitHub

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.

TensorFlow

tf

tf.audio

tf.autograph

tf.bitwise

tf.compat

tf.config

tf.data

tf.debugging

tf.distribute

tf.dtypes

tf.errors

tf.estimator

tf.experimental

tf.feature_column

tf.graph_util

tf.image

tf.initializers

tf.io

tf.keras

tf.linalg

tf.lite

tf.lookup

tf.losses

tf.math

tf.metrics

tf.nest

tf.nn

tf.optimizers

tf.quantization

tf.queue

tf.ragged

tf.random

tf.raw_ops

tf.saved_model

tf.sets

tf.signal

tf.sparse

tf.strings

tf.summary

tf.sysconfig

tf.test

tf.tpu

tf.train

tf.version

tf.xla

tf.estimator / experimental / experimental.Overview

View aliases

Args

Returns

Raises