【发布时间】:2017-07-30 13:20:47
【问题描述】:
我有一个由 3 个隐藏层组成的神经网络(所以我总共有 5 个层)。我想在每个隐藏层使用 Rectified Linear Units,但在最外层我想在 logits 上应用 Softmax。我想使用DNNClassifier。我已经阅读了 TensorFlow 的 official documentation,其中设置参数 activation_fn 的值他们说:
activation_fn:应用于每一层的激活函数。如果没有,将使用 tf.nn.relu。
我知道我总是可以编写自己的模型并使用激活函数的任意组合。但由于DNNClassifier 更具体,我想诉诸于此。到目前为止,我有:
classifier = tf.contrib.learn.DNNClassifier(
feature_columns=features_columns,
hidden_units=[10,20,10],
n_classes=3
# , activation_fn:::: I want something like below
# activation_fn = [relu,relu,relu,softmax]
)
【问题讨论】:
标签: python tensorflow neural-network softmax activation-function