如何在tensorflow r0.9中更改DNNClassifier中的激活功能？

Question

我找不到在DNNClassifier中更改激活功能的方法。该文件写得不好。我想做点什么：

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

但是在功能上没有激活_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}