Method __init__

train.py:24–25

(self, features)

class myDataset(Dataset):
    def __init__(self, features):
        self.features = features

    def __getitem__(self, idx):
        return self.features[idx]