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