【发布时间】:2019-02-25 19:01:18
【问题描述】:
我想训练一个模型,为可变大小的事件序列分配一些分数。每个序列都在自己的文件中,我从(文件名,目标分数)列表开始。
所以我做了这样的事情:
dataset = fileNames.map((fileName, score) => (new CsvDataset(x), score));
我得到的是:NotImplementedError : The Dataset.map() transformation does not currently support nested datasets as outputs
我使用的是 TensorFlow 1.10。
问题是:如何将(序列、训练分数)对加载并提供给模型? tf.data 是一种可行的方法吗?
【问题讨论】:
-
显然不清楚,但
fileNames已经是Dataset。 -
那么你可以使用
tf.data.Dataset.zip((dataset_x, dataset_y)) -
问题不在于我不知道如何压缩两个数据集。问题是将包含 (1,) 张量(包含 csv 文件名)的
Dataset转换为包含数据集的Dataset,其中单个嵌套的Dataset包含相应文件的行。似乎.map不支持。我应该将每个.csv文件转换为Tensor吗? -
您可以在您使用的任何数据集形状上使用
tf.decode_csv。编辑后的答案,希望我理解正确。
标签: python tensorflow keras tensorflow-datasets