在keras中创建自定义激活功能

Question

我想用keras.backend为LSTM的最后一个密写这样的自定义激活函数：

def customactivation(x):
    if x <= 0.5:
        return 0
    else :
        return 1



model.add(Dense(1, activation=customactivation))

我该怎么办？

Answer 1

该功能是不可区分的，除非您知道自己在做什么，否则它将对培训无用。您将收到错误“操作对梯度没有任何作用”

说：

def customactivation(x):
    return K.cast(K.greater(x, 0.5), K.floatx())