I m new to tensorflow and i need some help pls

I have a dataset that i downloaded from kaggle and it s in 4 folders trainA,trainB,testA,testB And i would like to put these image folders into a tf.data.dataset called DATA for example and i would like to call DATA[‘trainA’] for example to get the first folder thank you