Method __init__

mlh/data_preprocessing/data_loader.py:417–419

(self, dataset, indices=None)

class IndicesDataset:
    def __init__(self, dataset, indices=None):
        self.dataset = dataset
        self.indices = range(len(dataset)) if (indices is None) else indices

    def __getitem__(self, i):
        return self.dataset[self.indices[i]]