data_augmentation = tf.keras.Sequential([ layers.RandomRotation(0.2), layers.RandomFlip('horizontal_and_vertical'), ])