Как написать функцию линейной активации в Keras - PullRequest
1 голос
/ 25 мая 2020

Я хочу использовать линейную активацию на моем выходном слое. В настоящее время я использую model.add(...) для добавления слоя. Я знаю, как использовать другую функцию активации (например, сигмовидную функцию, вы можете использовать model.add(...,activation = "sigmoid"). Но мой вопрос в том, что мне просто не нужна функция активации. То есть я хочу w • x в качестве моего вывод модели вместо ( w • x ), где a - функция активации. Что мне делать? Будет просто не включать activation работу?

1 Ответ

2 голосов
/ 25 мая 2020

Создайте свою собственную функцию активации, которая возвращает то, что нужно.

from keras.utils.generic_utils import get_custom_objects
from keras.layers import Activation

def custom_activation(x):
    return x

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

model.add(...,activation = "custom_activation")

Редактировать

Как указано @MarcoCerliani, делать что-то выше не нужно. Ниже в обоих операторах используется функция активации linear.

model.add(...,activation = "linear")

model.add(...)
...