从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进行解析。具体例子可以看第四章。应该能满足你的需求。

0
3
正十七
回复
慕粉2767861
是不是可以尝试把40个二分类组织成一个向量,然后想分什么类就用对应的索引就可以?
2019-10-24
共3条回复

Google老师亲授 TensorFlow2.0 入门到进阶

Tensorflow2.0实战—以实战促理论的方式学习深度学习

1849 学习 · 896 问题

查看课程