tf.data.dataset pilih file dengan label filter

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