for a GAN model, I have an image dataset folder splitted between train, test and validation sub-folders. In each sub-folder 4 classes (1 Normal and 3 abnormal patients). How can I split this dataset to train my model only on normal patients and test the model on unseen normal and abnormal patients?