向tf.data.dataset添加“列”的最佳方法

import tensorflow as tf

# 定义一个函数来添加新的列
def add_column(element):
    new_element = {
        'feature1': element['feature1'],
        'feature2': element['feature2'],
        'new_column': element['feature1'] + element['feature2']  # 添加新的列
    }
    return new_element

# 创建一个示例数据集
dataset = tf.data.Dataset.from_tensor_slices({
    'feature1': [1, 2, 3, 4, 5],
    'feature2': [6, 7, 8, 9, 10]
})

# 使用map()函数添加新的列
new_dataset = dataset.map(add_column)

# 打印新的数据集
for element in new_dataset:
    print(element)