如何在 tensorflow r0.9 中更改 DNNClassifier 中的激活函数？

Question

我找不到在 DNNClassifier 中更改激活函数的方法。文件写得不好。我想做类似的事情：

  classifier = learn.DNNClassifier(hidden_units=[8,16,8], n_classes=2, activation_fn=relu)

但函数中没有activation_fn，所以我几乎无法更改它。

有人可以帮忙吗？谢谢，

Answer 1

所以有很多不同的激活函数。下面的字典只为您提供更常见的字典。你可以在这里找到所有激活函数：https://www.tensorflow.org/versions/r0.11/api_docs/python/nn.html

activation = {'relu': tf.nn.relu,
          'tanh': tf.nn.tanh,
          'sigmoid': tf.nn.sigmoid,
          'elu': tf.nn.elu,
          'softplus': tf.nn.softplus,
          'softsign': tf.nn.softsign,
          'relu6': tf.nn.relu6,
          'dropout': tf.nn.dropout}