当我们使用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)
对此我有两个问题。
使用tf.keras构建模型时,上述方法是否正确?
如果文件中有两个模型,并且只想为其中一个设置随机种子,那我该怎么办?
谢谢!