Method shuffle

cogdl/data/sampler.py:59–60

(self)

        return data[0]

    def shuffle(self):
        self.dataset.shuffle()

class NeighborSamplerDataset(torch.utils.data.Dataset):
    def __init__(self, dataset, sizes: List[int], batch_size: int, mask=None):