如何加载加权分割tensorflow数据集

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

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

Answer 2

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

import tensorflow_datasets as tfds

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