def get_training_dataset(): lst = [flip, rotate, color] dataset = load_dataset(TRAINING_FILENAMES, labeled=True) for data_augment in lst: dataset = dataset.map(dat
