Создать пользовательскую функцию активации в керасе

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())