【发布时间】:2019-11-04 16:36:03
【问题描述】:
可以为 tensorflow 的 DNNClassifier 罐装估计器自定义输出层的激活函数,但没有关于隐藏层默认激活函数的文档...是 ReLu 吗?可以定制吗?
【问题讨论】:
-
不只是输出层吗??
标签: tensorflow-estimator activation-function
可以为 tensorflow 的 DNNClassifier 罐装估计器自定义输出层的激活函数,但没有关于隐藏层默认激活函数的文档...是 ReLu 吗?可以定制吗?
【问题讨论】:
标签: tensorflow-estimator activation-function
您可以通过在实例化DNNClassifier 时为activation_fn 参数分配新的激活函数来修改隐藏层的激活函数。
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")
【讨论】: