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 не является целым числом. |
