Кои методи трябва да се реализират при наследяване от класа Dataset в PyTorch?

При наследяване от класа Dataset в PyTorch е необходимо да реализирате поне два метода:

• __len__(self) — връща броя на елементите в набора от данни.
• __getitem__(self, idx) — връща елемента с индекс idx.

Пример:

from torch.utils.data import Dataset

class MyDataset(Dataset):
    def __init__(self, data):
        self.data = data

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

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

Тези методи позволяват на DataLoader и други компоненти на PyTorch да работят правилно с вашия набор от данни.