从训练数据中随机抽取一打数据的好方法

  我们在模型训练的时候,每次分minbatch个数据进行训练,首先我们要随机地从中随机选取这么多个数据,有没有一套好的封装函数能够处理呢?有。我平时就遇到好多需要shuffle的地方,今天就总结一下这个函数,方便以后使用,提供给大家。直接上代码吧:

复制代码
 1 def chunks(n, *args):
 2     """Yield successive n-sized chunks from l. 从数据中连续地产生n个块
 3     输入参数:
 4         n:minbatch的大小,就是你需要一次从数据源里面拿多少数据出来
 5         *args:其它的一些参数,包括数据长度索引、源数据等
 6     输出:
 7         shuffle的索引和数据
 8     """
 9     # 存放shuffle索引的列表
10     keypoints = []
11     for i in range(0, len(args[0]), n):
12         keypoints.append((i, i + n))
13     random.shuffle(keypoints)
14     # 获取随机生成的索引值和数据值,通过函数的每次调用返回
15     for a, b in keypoints:
16         # 每次生成两个list,第一个是索引值,第二个是对应的数据值
17         yield [arg[a: b] for arg in args]
复制代码

 

posted @   今夜无风  阅读(260)  评论(0编辑  收藏  举报
编辑推荐:
· 记一次.NET内存居高不下排查解决与启示
· 探究高空视频全景AR技术的实现原理
· 理解Rust引用及其生命周期标识(上)
· 浏览器原生「磁吸」效果!Anchor Positioning 锚点定位神器解析
· 没有源码,如何修改代码逻辑?
阅读排行:
· 分享4款.NET开源、免费、实用的商城系统
· 全程不用写代码,我用AI程序员写了一个飞机大战
· MongoDB 8.0这个新功能碉堡了,比商业数据库还牛
· 白话解读 Dapr 1.15:你的「微服务管家」又秀新绝活了
· 上周热点回顾(2.24-3.2)
历史上的今天:
2017-09-29 sklearn学习_01
2017-09-29 python 关于GUI的思考(转)
点击右上角即可分享
微信分享提示