从tfrecord读取数据的多输出的写法
来源:1-7 Google_cloud无GPU环境搭建
慕粉2767861
2019-10-19
老师您好,我尝试从 tfrecord 中读取数据,进行多输出的训练。因为每一个输出有不一样的label,所以需要在fit 的时候逐个output对应label。
问题是:从tfrecord中解析出的是一个dataset对象,我没法通过字典的形式 [ ‘name’],取出其中一项。请问有什么解决的方法吗?
写回答
1回答
-
正十七
2019-10-20
同学你好,你可以在存储tfrecord文件的时候使用tf.Example存储,然后解析的时候也按照tf.Example进行解析。具体例子可以看第四章。应该能满足你的需求。
032019-10-24
相似问题
keras读入数据的方式
回答 1
自己的数据集
回答 1