tf.keras.models中的随机种子

时间:2019-09-05 22:25:06

标签: tensorflow random keras seed

当我们使用tf.keras构建模型时,我希望输出具有一致的值。也就是说,我不希望输出随机变化。例如,如果我运行以下代码,则输出始终是不同的,并且我猜这可能是由于参数的随机初始化所致。

inputs1 = tf.keras.layers.Input(shape=(3, 1)) 
lstm1 = tf.keras.layers.LSTM(1)(inputs1)
model = tf.keras.models.Model(inputs = inputs1, outputs = lstm1)

data  = np.array([0.1, 0.2, 0.3]).reshape(1, 3, 1)
print (model.predict(data))

要解决此问题,我注意到在文件开头添加以下行可以解决此问题。

tf.random.set_random_seed(0)

对此我有两个问题。

  1. 使用tf.keras构建模型时,上述方法是否正确?

  2. 如果文件中有两个模型,并且只想为其中一个设置随机种子,那我该怎么办?

谢谢!

0 个答案:

没有答案