具有阈值替换功能的keras自定义激活功能

Question

对于要更改为低于分数的自定义激活功能，我想将其值替换为Activated_x，将其小于threshold = 0.5的值设为0。

如何修改？

def ScoreActivationFromSigmoid(x, target_min=1, target_max=9) :
    activated_x = K.sigmoid(x)
    #threshold = 0.5
    #binary_activated_x = activated_x > threshold
    #activated_x = K.cast_to_floatx(binary_activated_x)
    score = activated_x * (target_max - target_min) + target_min
    return  score

使用上面注释掉的源代码，它可以用作激活功能，但是当我取消注释时，它们将不起作用

Answer 1

方法K.cast_to_floatx适用于numpy数组，而不适用于张量。您可以改为使用函数K.cast，如下所示：

activated_x = K.cast(binary_activated_x, dtype='float32')