TFReocrd类型数据读写

acnowa

于 2021-05-27 19:37:23 发布

阅读量168

点赞数

分类专栏： python 文章标签： python

本文链接：https://blog.csdn.net/qq_38081870/article/details/117336012

版权

本文介绍了两种在Python中操作TFRecord文件的方法，包括数据的遍历与读取，适用于处理大规模机器学习数据集。

摘要由CSDN通过智能技术生成

这里结合自身实际使用，记录一下对TFRecord文件中每一条数据遍历的方法。

方法一

该方法在自己的测验过程感觉比较快一点

    cnt = 0
    for serial_exp in tf.python_io.tf_record_iterator('../data/20_part-r-00199.tfrecord'):
        example = tf.train.Example()
        example.ParseFromString(serial_exp)
        label = example.features.feature['label'].int64_list.value
        feat = example.features.feature['id'].int64_list.value
        wt = example.features.feature['wt'].float_list.value
        if cnt == 0:
            print(label)
            print(feat)
            print(wt)