在 keras 中创建自定义激活函数

Question

我正在尝试在 keras 中创建自己的自定义激活函数，如果 x < 0 则返回 0，如果 x >= 0 则返回 1

它给了我这个错误

我该如何解决？

Answer 1

警告：您想要的此操作没有梯度，并且在可训练之前不允许任何权重。 您将看到诸如“操作没有渐变”或“不支持无类型”之类的错误消息。

作为激活的变通方法，我相信“relu”激活将是最接近和最好的选择，其优点是非常受欢迎并在大多数模型中使用。

在 Keras 中，您通常不会运行会话。 对于自定义操作，您可以使用后端函数创建一个函数。

因此，您将使用Lambda层：

import keras.backend as K

def hardlim(x):
   return K.cast(K.greater_equal(x,0), K.floatx())

然后，您可以在层中使用activation=hardlim 。