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

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

tensorflow-datasets tensor tensorflow2.0 keras tensorflow

Marsellus Wallace · 技术社区 · 5 年前

我有一个 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 回复 | 直到 5 年前

today 5 年前

您可以定义一个函数,该函数可以接受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)

推荐文章

albert · AttributeError:“str”对象没有属性“str”如何解决此问题

3 年前

CoccoSyn · Keras TypeError:其中一个输入没有可接受的类型。具有参差不齐张量的LSTM

3 年前

Joseph · 重塑BatchDataset训练模型的输入-Tensorflow

3 年前

Chris Ranck · Keras时间序列生成器-输入为“X”长,输出为“X-长”

3 年前

DPM · 将自定义掩码传递给LSTM数据以进行培训和验证

3 年前

Luciana Carota · 为什么Keras的R“预测”函数不能识别正确的输入维度?

3 年前

Sarah Elnaggar · 我使用Gensim Doc2vec进行图形嵌入,然后在keras中使用两层深度神经网络进行二元分类

3 年前

user836026 · 在pytorch中复制权重时语法无效

3 年前

mad · 如何从一个定制的神经网络模型中获得逻辑和概率

3 年前

Archit · 无法使用自定义距离函数加载带有lambda函数的keras模型

7 年前