Welke methoden moeten worden geïmplementeerd bij het erven van de Dataset-klasse in PyTorch?

Bij het erven van de klasse Dataset in PyTorch, moet je minimaal twee methoden implementeren:

• __len__(self) — geeft het aantal elementen in de dataset terug.
• __getitem__(self, idx) — geeft het element met index idx terug.

Voorbeeld:

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]

Deze methoden stellen DataLoader en andere PyTorch-componenten in staat om correct met je dataset te werken.