学堂 学堂 学堂公众号手机端

Keras中如何设置随机种子

lewis 2年前 (2023-09-29) 阅读数 7 #技术

在Keras中,可以通过设置numpytensorflow的随机种子来控制模型的随机性。

import numpy as np
import tensorflow as tf

# 设置 numpy 的随机种子
np.random.seed(0)

# 设置 tensorflow 的随机种子
tf.random.set_seed(0)

在创建模型之前,使用以上代码设置随机种子可以确保每次运行时生成的随机数相同,从而保证模型的可重复性。


版权声明

本文仅代表作者观点,不代表博信信息网立场。

热门