tf.compat.v1.estimator.experimental.dnn_logit_fn_builder

Function builder for a dnn logit_fn.

tf.compat.v1.estimator.experimental.dnn_logit_fn_builder(
    units, hidden_units, feature_columns, activation_fn, dropout,
    input_layer_partitioner, batch_norm
)

Args
units	An int indicating the dimension of the logit layer. In the MultiHead case, this should be the sum of all component Heads' logit dimensions.
hidden_units	Iterable of integer number of hidden units per layer.
feature_columns	Iterable of feature_column._FeatureColumn model inputs.
activation_fn	Activation function applied to each layer.
dropout	When not None, the probability we will drop out a given coordinate.
input_layer_partitioner	Partitioner for input layer.
batch_norm	Whether to use batch normalization after each hidden layer.

Returns
A logit_fn (see below).

Raises
ValueError	If units is not an int.