TensorFlow 数据处理:TFRecord 与特征预处理
1. 压缩 TFRecord 文件
在处理 TFRecord 文件时,压缩操作有时非常有用,特别是当需要通过网络连接加载文件时。以下是创建和读取压缩 TFRecord 文件的具体步骤:
1.1 创建压缩 TFRecord 文件
可以通过设置 options 参数来创建压缩的 TFRecord 文件,示例代码如下:
import tensorflow as tf
options = tf.io.TFRecordOptions(compression_type="GZIP")
with tf.io.TFRecordWriter("my_compressed.tfrecord", options) as f:
# 这里可以添加写入具体数据的代码
pass
1.2 读取压缩 TFRecord 文件
在读取压缩的 TFRecord 文件时,需要指定压缩类型,示例代码如下:
dataset = tf.data.TFRecordDataset(["my_compressed.tfrecord"],
compression_type="GZIP")
2. Protocol Buffers 简介
TFRecord 文件通常包含序列化的
超级会员免费看
订阅专栏 解锁全文
15

被折叠的 条评论
为什么被折叠?



