I\'m new to Tensorflow and i\'m trying to train a model for which I am using an ImageDataGenerator:
data_gen = tf.keras.preprocessing.image.ImageDataGenerator