以设定的比例混合数据集

【问题标题】：Mixing datasets in set ratio以设定的比例混合数据集
【发布时间】：2019-04-11 18:37:12
【问题描述】：

在 tensorlfow 数据集中，我如何混合 2 个数据集，从原始数据中获取 75% 的数据集，从增强数据中获取 25% 的数据集？

d = tf.data.Dataset.list_files("raw_data/")\
    .flat_map(tf.data.TFRecordDataset)
ad = tf.data.Dataset.list_files("augmented_data/")\
    .flat_map(tf.data.TFRecordDataset)

【问题讨论】：

标签： tensorflow tensorflow-datasets tensorflow-estimator

【解决方案1】：

问题是您不能在数据集对象上使用len()，因此在迭代一个完整的时期之前，有时很难知道示例的确切数量。但是你可以用take 和skip 方法来近似。

train_dataset = dataset.take(number_examples_for_train)
test_dataset = dataset.skip(number_examples_for_train)

这些方法可以直接替代。 https://www.tensorflow.org/api_docs/python/tf/data/Dataset#take

【讨论】：