Add new activation functions
mehrdad-dev opened this issue · 2 comments
mehrdad-dev commented
How can I add an activation function like tanh for the "output_activation" parameter?
@yingkaisha
Thanks!
akaiml commented
Hi @mehrdad-dev, Did you get to know how to add tanh as an output activation function?
mehrdad-dev commented
you should change the code directly!
like this, and then pass your activation function
if activation:
if activation == 'Sigmoid':
X = Activation('sigmoid', name='{}_activation'.format(name))(X)
else:
X = activation(X)