Method __init__

code/utils.py:300–312

(self, original_dataset)

    
class OneShotDataset(Dataset):
    def __init__(self, original_dataset):
        self.data = []
        self.labels = []
        class_samples = {}

        for i in range(len(original_dataset)):
            data = original_dataset.data.x[i]
            label = original_dataset.data.y[i].item()
            if label not in class_samples:
                class_samples[label] = data
                self.labels.append(label)

        self.data = list(class_samples.values())

    def __len__(self):
        return len(self.data)