如何加载加权拆分张量流数据集

时间:2020-04-03 07:49:30

标签: python tensorflow tensorflow-datasets

我按此代码进行加权分割:

half, quarter, quarter2 = tfds.Split.TRAIN.subsplit(weighted = [2, 1, 1])

之后,我尝试通过以下方式加载拆分数据:

dataset = tfds.load('mnist', split=half)

但是我得到了错误:

AssertionError: Unrecognized instruction format: NamedSplit('train')(tfds.percent[0:50])

什么是正确的加载语法。

2 个答案:

答案 0 :(得分:1)

解决您问题的另一种方法是使用这种切片方式

import tensorflow_datasets as tfds

s1,s2,s3 = "train[:50]","train[:25]","train[:25]"
dataset= tfds.load('mnist', split=s1)

答案 1 :(得分:1)

api tfds.Split.TRAIN.subsplit已弃用。它不适用于tfds 2.1。峨山的解决方案是正确的。