PytorchのDatasetを層状にfoldingする - Qiita

既存のDatasetをN-foldするときに、先頭から順に1,2,...N,1,2,...,N,1,2,...,N,1,2,...とデータセットを分割する。時系列データを分割するときに、特定の月や季節にデータが偏らないようにするときに使用する。 from torch.utils.data import Dataset class LayeredFoldWrapper(Dataset): def __init__(self, dataset, n_splits=5, fold=0, valid=False): self.dataset = dataset self.n_splits = n_splits self.fold = fold self.valid = valid self.valid_index = list(self._valid_index(len(dataset), n_s