W3cubDocs

/TensorFlow 2.3

tf.compat.v1.estimator.experimental.dnn_logit_fn_builder

Function builder for a dnn logit_fn.

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.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/compat/v1/estimator/experimental/dnn_logit_fn_builder