Вы можете изменить функцию активации скрытого слоя, назначив новую функцию активации аргументу 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")