def __getitem__(self, index):
    data = self.selected_data[0][index]
    target = self.selected_data[1][index]
    if (self.transform is not None):
        data = self.transform(data)
    return (data, target)
def __getitem__(self, index):
    data = self.selected_data[0][index]
    target = self.selected_data[1][index]
    if (self.transform is not None):
        data = self.transform(data)
    return (data, target)