class SiameseDataset(Dataset):
    def __init__(self, data, transform=None):
        self.data = data    
        self.transform = transform