Использование подмножества tfrecord

Question

Можно ли использовать существующий tfrecord для одного или подмножества меток, которые использовались для его создания

Я тренирую несколько моделей с одинаковыми данными, для каждой из которых потребуется только одна или подмножество меток, использованных для первоначального создания tfrecord. tfrecord довольно большой, поэтому я хочу создать его для каждого поднабора моделей.

Answer 1

tf.data.Dataset s имеет filter, skip и take методы, которые вы можете найти полезными.В качестве альтернативы вы можете разделить свой исходный набор данных на несколько файлов tfrecord и создать Dataset на основе подмножества этих файлов.

Если вы готовы воссоздать данные, используя tensorflow_datasets, split может также дать вам то, что вы хотите.