python - 为调用 Keras 模型的类设置种子

Question

我有用于优化 Keras LSTM 模型参数的类。众所周知，要为 keras 设置种子，必须在其代码中输入以下内容。但我不明白的是，在将构建和修改模型的类的情况下，它应该放在哪里。

from numpy.random import seed
seed(1)
from tensorflow.random import set_seed
set_seed(2)

它应该在下面的 __init__ 中吗？

from numpy.random import seed
from tensorflow.random import set_seed

class OptimizeLSTM:
    def __init__(self, X_train, y_train, X_test, y_test, verbose=False):
        self._X_train = X_train
        self._y_train = y_train
        self._X_test = X_test
        self._y_test = y_test
        self._verbose = verbose
        seed(1)
        set_seed(2)

标签： pythonclasskerasseed