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

Question

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

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

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

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

但是我得到了错误：

...AssertionError：无法识别的指令格式：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)