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

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

tensorflow-datasets tensor tensorflow2.0 keras tensorflow

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 年前

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)

推荐文章

xposure · Tensorflow:训练神经网络时损失没有改善

6 年前

user179156 · 用于向tensorflow模型提供tfrecords字节的java api

6 年前

user25004 · 如何将Pytroch张量从数学上的列向量转换为列矩阵?

6 年前

sashaostr · Keras/Tensorflow中不同时期的训练率如何变化

6 年前

Tom · 如何重塑keras中的张量?

6 年前

Tom · 如何将tensorflow中的dropout应用于多维张量?

6 年前

Sakhri Houssem · 为什么我必须在美国有线电视新闻网(CNN)中将一幅图像重塑为[n,高度,宽度,频道]

6 年前

Jame · 如何打印张量的形状值?

6 年前

baptiste · 动态创建d维张量

6 年前

Brian · Tensorflow-按批次索引对占位符进行分组

7 年前