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

时间:2018-06-07 05:30:16

标签: tensorflow keras kernel initializer seed

我想在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)

如果我删除“(seed = 0)”,如下所示

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

在没有设定种子值的情况下效果很好。

如何设置种子值?

1 个答案:

答案 0 :(得分:4)

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