tf.data.dataset Seleccione archivos con filtro de etiquetas

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