首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >将tf.data.Dataset拆分为两个不同的输入和目标tf.data.Dataset

将tf.data.Dataset拆分为两个不同的输入和目标tf.data.Dataset
EN

Stack Overflow用户
提问于 2020-04-02 01:43:09
回答 1查看 355关注 0票数 2

如何从同时包含输入和目标数据的tf.data.Dataset中创建两个不同的输入和目标tf.data.Dataset,以便将它们用作model.fit函数的x和y参数?

我使用以下代码加载数据集

代码语言:javascript
复制
    dataset = tf.data.TFRecordDataset(
        dataset_file_path, 
        compression_type='GZIP', 
        buffer_size=None, 
        num_parallel_reads=None
    )
EN

回答 1

Stack Overflow用户

发布于 2020-04-02 03:04:05

您是如何创建此数据集的?

如果dataset中的每个元素都采用(input_data, target_data)格式,则可以执行以下操作

代码语言:javascript
复制
input_dataset = dataset.map(lambda x,y: x)
target_dataset = dataset.map(lambda x,y: y)
票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/60977051

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档