tf.compat.v1.estimator.experimental.dnn_logit_fn_builder
Функция-строитель для 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
)
| Аргументы |
units | Целое число, указывающее размерность слоя logit. В случае MultiHead это должно быть суммой размеров logit всех компонентных голов. |
hidden_units | Итерируемый набор целых чисел, количество скрытых узлов на каждом слое. |
feature_columns | Итерируемый набор feature_column._FeatureColumn входных данных модели. |
activation_fn | Функция активации, применяемая к каждому слою. |
dropout | Если не None, вероятность отбрасывания заданной координаты. |
input_layer_partitioner | Разделитель для входного слоя. |
batch_norm | Использовать ли пакетную нормализацию после каждого скрытого слоя. |
| Возвращает |
| Функция logit_fn (см. ниже). |
| Исключения |
ValueError | Если units не целое число. |