Сделать метод набора данных pytorch __getitem__ зависимым от параметров ⇐ Python

[Anonymous]

Я хотел бы работать с пользовательским классом набора данных Pytorch, в котором выходные данные метода getitem зависят от параметра.

Код: Выделить всё

class myDataset(Dataset):

def __init__(self, myparam, dataframe, dataframe2):
self. myparam = myparam
self.dataframe = dataframe
self.dataframe2 = dataframe2

def __getitem__(self, idx):

X = self.dataframe.iloc[idx].values
y = self.dataframe2.iloc[idx].values

if self.myparam == 1:
X = ... # manipulate X in some way
return X, y

dataset = myDataset(myparam, dataframe, dataframe2)
validation_loader = DataLoader(dataset, batch_size=batch_size, sampler=val_subsampler)

Проблема в том, что я хотел бы получить доступ к параметру Dataset из созданного загрузчика данных таким образом, чтобы я мог выбрать манипулировать или нет X данные непосредственно из объекта validation_loader
Для решения этой проблемы может быть несколько обходных путей, но мне был бы интересен подход «хорошего кода».

Подробнее здесь: https://stackoverflow.com/questions/786 ... parameters