将tf.data.Dataset分成两个不同的Input和Target tf.data.Dataset

Question

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

我使用下面的代码加载数据集

    dataset = tf.data.TFRecordDataset(
        dataset_file_path, 
        compression_type='GZIP', 
        buffer_size=None, 
        num_parallel_reads=None
    )

Answer 1

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

如果数据集中的每个元素的格式均为(input_data, target_data)，则可以进行

input_dataset = dataset.map(lambda x,y: x)
target_dataset = dataset.map(lambda x,y: y)