有没有办法为keras的一层中的每个隐藏单元设置不同的激活函数?

数据挖掘 机器学习 神经网络 深度学习 喀拉斯 激活函数
2021-09-28 17:18:27

我正在尝试为层中的每个隐藏单元设置不同的激活函数。这在 Keras 中是否可以使用“连接”?

1个回答

如果我明白了,您可以使用类似的代码,如下所示:

from keras.layers import merge, Convolution2D, MaxPooling2D, Input

input = Input(shape=(256, 256, 3))

seq1 = Dense(1, activation = 'relu')(input)
seq2 = Dense(1, activation = 'sigmoid')(input)
seq3 = Dense(1, activation = 'tanh')(input)

acum = merge([seq1, seq2, seq3], mode='concat', concat_axis=1)

根据您的任务,指定concat_axis.