Как создать пользовательскую функцию активации с Keras?

иногда по умолчанию стандартные активаций как Релу, Танха, softmax, ... и дополнительно активаций как LeakyReLU не достаточно. И это также может быть не в keras-contrib.

Как вы создаете свою собственную функцию активации?

11
keras keras-layer python

1 ответов

кредиты этот комментарий к выпуску Github от Ritchie ng.

from keras.layers import Activation
from keras import backend as K
from keras.utils.generic_utils import get_custom_objects

def custom_activation(x):
    return (K.sigmoid(x) * 5) - 1

get_custom_objects().update({'custom_activation': Activation(custom_activation)})

model.add(Activation(custom_activation))

пожалуйста, имейте в виду, что вы должны импортировать эту функцию при сохранении и восстановлении модели. См.записка кераса-контриба.

27