data_loader.py 文件源码

def load_test_data(self):
        # Remove non-mat files, and perform ascending sort
        allfiles = os.listdir(self.data_dir)
        npzfiles = []
        for idx, f in enumerate(allfiles):
            if ".npz" in f:
                npzfiles.append(os.path.join(self.data_dir, f))
        npzfiles.sort()

        # Files for validation sets
        val_files = np.array_split(npzfiles, self.n_folds)
        val_files = val_files[self.fold_idx]

        print "\n========== [Fold-{}] ==========\n".format(self.fold_idx)

        print "Load validation set:"
        data_val, label_val = self._load_npz_list_files(val_files)

        return data_val, label_val