Разделите набор tf.data.Datase на два различных входа и целевой объект tf.data.Dataset.

Question

Как я могу создать два отличных Input и Target tf.data.Dataset от tf.data.Dataset, содержащего и Input, и Target data, чтобы я мог использовать их как параметры x и y функции model.fit?

Я использую приведенный ниже код для загрузки набора данных

    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)