数据集封装
来源: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。
00
相似问题