Функция активации для тензорного потока скрытых слоев DNNCalssifier - PullRequest
0 голосов
/ 04 ноября 2019

Возможно настроить функцию активации выходного слоя для стандартного оценщика DNNClassifier от tenorflow, но нет документации о функции активации скрытого слоя по умолчанию ... Это ReLu? Это настраиваемо?

1 Ответ

0 голосов
/ 05 ноября 2019

Вы можете изменить функцию активации скрытого слоя, назначив новую функцию активации аргументу activation_fn при создании экземпляра DNNClassifier.

classifier = tf.estimator.DNNClassifier(feature_columns=feature_columns,
                                        hidden_units=[10, 20, 10],
                                        activation_fn=tf.nn.sigmoid,  # default value is relu
                                        n_classes=3,
                                        model_dir="/tmp/model")
...