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

Question

我通过以下代码进行加权拆分：

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])

加载的正确语法是什么。

Answer 1

问题的替代解决方案是使用这种类型的切片

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

Answer 2

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