Как установить начальное значение инициализатора ядра (glorot_uniform) в Keras

Я хотел бы установить начальное значение инициализатора ядра glorot_uniform в Keras.

model.add(Dense(50, input_dim=self.state_size, activation='relu', kernel_initializer='glorot_uniform(seed=0)'))

Когда я использую приведенный выше код, сообщение об ошибке ниже.

ValueError: Unknown initializer: glorot_uniform(seed=0)

Если я удалю "(seed=0)", как показано ниже

 model.add(Dense(50, input_dim=self.state_size, activation='relu', kernel_initializer='glorot_uniform'))

Он хорошо работает без установки начального значения.

Как установить начальное значение?


person WKK    schedule 07.06.2018    source источник


Ответы (1)


Keras может использовать строки и функции в качестве аргументов для инициализаторов. Строки просто используют параметры по умолчанию для инициализаторов. Попробуйте эту строку кода для вашего слоя FC:

from keras import initializers
model.add(Dense(50, input_dim=self.state_size, activation='relu', kernel_initializer=initializers.glorot_uniform(seed=0)))

Здесь у вас есть документация для инициализаторов: https://keras.io/initializers

person Lau    schedule 07.06.2018