if split == 'training':
            # Training split folder and file with data question.
            data_file = os.path.join(self.data_folder, 'ChemDATA_A_Dist_Labels_Set0.tsv')
            self.image_folder = os.path.join(self.data_folder, "ChemDATA_A_Dist")