美文网首页
data_iter(batch_size, features,

data_iter(batch_size, features,

作者: elementmaster | 来源:发表于2020-03-12 16:01 被阅读0次

defdata_iter(batch_size,features,labels):

    num_examples=len(features)

    indices=list(range(num_examples))

    random.shuffle(indices)   

   for i in range(0,num_examples,batch_size):

            j=nd.array(indices[i:min(i+batch_size,num_examples)])

            yield features.take(j),labels.take(j)            # take函数根据索引返回对应元素

相关文章

网友评论

      本文标题:data_iter(batch_size, features,

      本文链接:https://www.haomeiwen.com/subject/hoypjhtx.html