如何在 Keras 中设置内核初始值设定项 (glorot_uniform) 的种子值

Question

我想在 Keras 中设置glorot_uniform内核初始值设定项的种子值。

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

当我使用上面的代码时，错误消息如下。

ValueError: Unknown initializer: glorot_uniform(seed=0)

如果我删除"(种子=0("，如下所示

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

无需设置种子值即可正常工作。

如何设置种子值？

Answer 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