如何通过 TensorFlow Transform 管道传递 TensorFlow 数据集？

编程问答 2022-07-16

问题描述

我已经为我的原始数据实现了一个 custom TensorFlow Dataset。我可以将数据作为 tensorflow.data.Dataset 下载、准备和加载，如下所示：

import tensorflow_datasets

builder = tensorflow_datasets.builder("my_dataset")
builder.download_and_prepare()
ds = builder.as_dataset()

我想在 TensorFlow Transform pipeline 中转换此数据以进行模型训练。但是，我能够将数据集传递到转换管道的唯一方法是将其转换为实例字典并传入原始数据元数据。

instance_dicts = tensorflow_datasets.as_dataframe(ds).to_dict(orient="records")
with tensorflow_transform.beam.Context():
    (transformed_data,_),transform_fn = (
        instance_dicts,RAW_DATA_MetaDATA,) | tensorflow_transform.beam.AnalyzeAndTransformDataset(
        preprocessing_fn,output_record_batches=True
    )

是否有一种更简单、更高效的方式将 TensorFlow 数据集传递到 TensorFlow 转换管道？

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）

pythontensorflowtensorflowtensorflowtensorflow-datasetstensorflow-transform