tf.data.dataset выберите файлы с мечениями фильтра

dataset = train_data.filter(lambda x,y: tf.reduce_all(tf.not_equal(y, [0,1,2]))).batch(200)
Handsome Hamerkop