微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

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

如何解决为调用 Keras 模型的类设置种子

我有用于优化 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)

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。