2016-07-20 36 views
0

我找不到在DNNClassifier中更改激活功能的方法。該文件寫得不好。我想要做的事,如:如何改變張量流r0.9中DNNClassifier的激活函數?

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

但是在溫控功能沒有activation_fn,所以我根本無法改變它。

誰能幫助?謝謝,

回答

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} 
+0

有沒有辦法使用不同的激活功能,每層按程序員的需要? – user110327

相關問題