tf.contrib.estimator.dnn_logit_fn_builder
Function builder for a dnn logit_fn.
tf.contrib.estimator.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. |
© 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/r1.15/api_docs/python/tf/contrib/estimator/dnn_logit_fn_builder