数据集封装

来源:7-12 数据集封装

慕九州3352676

2020-02-18

老师您好!

我想问:在神经网络训练中,数据集的封装一般需要实现什么功能呀,一般提供哪些方法呢?

写回答

1回答

正十七

2020-02-24

你可以参考tensorflow中新的关于dataset的API:https://www.tensorflow.org/api_docs/python/tf/data/Dataset

里面的功能主要有,随机混排shuffle,batch,拼接,样本变换map等等。如果是分布式的还需要支持分布式。

map这个函数比较通用,里面的参数是一个函数,可以接受一个样本,返回一个变换后的新样本。

如果是自己实现的话,一般都是单机的环境,需要的方法就是返回下一个batch。

0
0

深度学习之神经网络(CNN/RNN/GAN)算法原理+实战

深度学习算法工程师必学,深入理解深度学习核心算法CNN RNN GAN

2617 学习 · 935 问题

查看课程