Как загрузить набор данных взвешенного расщепленного тензорного потока

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 устарел. Это не работает с TFDS 2.1. Eshan решение правильное.

Answer 2

Альтернативным решением вашей проблемы было использование этого типа нарезки

import tensorflow_datasets as tfds

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