代码之家 › 专栏 › 技术社区 › 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)

推荐文章

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

2 年前

Anshuman Sinha · 将tensforflow中的所有nan值转换为零

2 年前

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

2 年前

rob me of my freedom · 在不使用Y_pred和Y_actual的tensorflow中编写自定义损失

2 年前

Chris · 如何捕获导入脚本功能的打印

2 年前

Jay Bhie Santos · 剩余GRU:ValueError:输入具有不兼容的形状。收到的形状(5)和(24)

2 年前

Danijel · C中TensorFlow模型文件的路径?

2 年前

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

2 年前

Arya Anish · tensorflow安装中面临的问题

2 年前

so beautiful memory · 如何将我在本地训练的tensorflow模型正确部署到谷歌云人工智能平台?我部署了它,但没有图像返回

2 年前