代码之家 › 专栏 › 技术社区 › Marsellus Wallace

如何将TensorFlow数据集列合并到Keras的(features,label)元组中?

tensorflow-datasets tensor tensorflow2.0 keras tensorflow

0

Marsellus Wallace · 技术社区 · 4 年前

我有一个 tf.data.Dataset 由多个字段组成(例如feature1、feature2、feature3、label)。如何将其转换为 Tensor Keras 为了模特训练?

更新

这个是一本类似的词典:

{
    'feature1': 1.5,
    'feature2': 5.6,
    'feature3': 7.9,
    'label': 1.0
}

预期的 煤油 输入格式是 tuple (features_tensor, label_tensor) .

0 回复 | 直到 4 年前

1

today 4 年前

您可以定义一个函数,该函数可以接受dictionary元素并以所需格式将数据作为元组返回。然后,你可以使用 tf.data.Dataset.map 方法将其应用于数据集:

def reformat_element(elem):
    features = [elem['feature1'], elem['feature2'], elem['feature3']]
    label = [elem['label']]
    return features, label

dataset = dataset.map(reformat_element)