Batch generator 批生成器(附代码)

深度学习中经常用到的一个技巧是使用批训练,这样的好处是可以减少显存的资源占用,对训练的结果也有一定的影响。 下面简单编写一个批量数据生成器:import random import numpy mode = 0x = np.arange(100)### 假设这个为featuresy = np.arange(100) ########## 假设这个为labelsdef ba...
摘要由CSDN通过智能技术生成

深度学习中经常用到的一个技巧是使用批训练,这样的好处是可以减少显存的资源占用,对训练的结果也有一定的影响。
下面简单编写一个批量数据生成器:

import random 
import numpy 
mode = 0

x = np.arange(100)### 假设这个为features
y = np.arange(100)  ########## 假设这个为labels

def batch_generator(data,shuffle,batch_size):
    count = 0
    if shuffle:  ### 是否打乱
        shuffle_index = random.shuffle(list(range(data[0].shape[0])),data[
  • 0
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值