data_loader.py 文件源码

def load(self):

        with open(os.path.join(self.data_dir, "notMNIST.pickle"), 'rb') as f:
            save = pickle.load(f)
            train_dataset = save['train_dataset']
            train_labels = save['train_labels']
            valid_dataset = save['valid_dataset']
            valid_labels = save['valid_labels']
            test_dataset = save['test_dataset']
            test_labels = save['test_labels']
            del save  # hint to help gc free up memory

        train_dataset, train_labels = self.reformat(train_dataset, train_labels, self.image_size, self.num_labels)
        valid_dataset, valid_labels = self.reformat(valid_dataset, valid_labels, self.image_size, self.num_labels)
        test_dataset, test_labels = self.reformat(test_dataset, test_labels, self.image_size, self.num_labels)

        return train_dataset, train_labels, valid_dataset, valid_labels, test_dataset, test_labels