如何在 Jupyter Notebook 中直接设置全局随机种子的方法及易错地方、notebook和pycharm中设置随机种子的区别

结论:

在 Jupyter Notebook 中直接设置全局随机种子的方法是确保每个单独的代码块中都调用相同的 set_seed 函数。这是最简单且有效的方法。在每个代码块开头设置随机种子,确保代码在每次执行时具有相同的随机数生成顺序。

易错地方

jupyter-notebook中:

直接在代码块的其中一个运行set_seed(2024),以为就全局可用了,实际上是只在当前代码块生效 。而notebook中是由很多个代码块构成的,所以每个代码块如果存在随机生成或者划分数据的函数,都应该在开头执行一下下面这行代码:

复制代码
set_seed(2024)

pycharm中:

在这个py脚本中,执行一次上面这个函数应该是随机种子全局有效的,因为py脚本只有一个代码块,不能分块执行。

其中,set_seed()函数如下所示,可以放在文件代码块的第一个块,方便后面的每个块调用

python 复制代码
import numpy as np
import torch
import random

def set_seed(seed):
    random.seed(seed)
    np.random.seed(seed)
    torch.manual_seed(seed)
    if torch.cuda.is_available():
        torch.cuda.manual_seed(seed)
        torch.cuda.manual_seed_all(seed)
    torch.backends.cudnn.deterministic = True
    torch.backends.cudnn.benchmark = False

代码验证测试如下:

相关推荐
F_D_Z3 小时前
哈希表解Two Sum问题
python·算法·leetcode·哈希表
智算菩萨3 小时前
【实战】使用讯飞星火API和Python构建一套文本摘要UI程序
开发语言·python·ui
Groundwork Explorer3 小时前
异步框架+POLL混合方案应对ESP32 MPY多任务+TCP多连接
python·单片机
梦帮科技3 小时前
Scikit-learn特征工程实战:从数据清洗到提升模型20%准确率
人工智能·python·机器学习·数据挖掘·开源·极限编程
xqqxqxxq3 小时前
Java 集合框架之线性表(List)实现技术笔记
java·笔记·python
verbannung3 小时前
Python进阶: 元类与属性查找理解
python
jugt3 小时前
将JetBrains系软件(PhpStorm、IntelliJIdea、WebStorm)缓存目录迁移到其它盘
ide·phpstorm
想用offer打牌3 小时前
LLM参数: Temperature 与 Top-p解析
人工智能·python·llm
小智RE0-走在路上4 小时前
Python学习笔记(6)--列表,元组,字符串,序列切片
笔记·python·学习
feeday4 小时前
Python 删除重复图片 优化版
开发语言·python