PyTorch DataLoader
Świetna robota z definicją klasy Dataset! Utworzony przez ciebie WaterDataset jest już gotowy do użycia.
Kolejnym krokiem w przygotowaniu danych treningowych jest skonfigurowanie DataLoader. PyTorchowy DataLoader można utworzyć na podstawie Dataset, aby wczytywać dane, dzielić je na partie (ang. batches) i w razie potrzeby stosować przekształcenia. Następnie generuje on próbki danych gotowe do trenowania.
W tym ćwiczeniu zbudujesz DataLoader oparty na WaterDataset. Klasa DataLoader, której będziesz potrzebować, została już zaimportowana z torch.utils.data. Do dzieła!
To ćwiczenie jest częścią kursu
Głębokie uczenie z PyTorch – poziom średnio zaawansowany
Instrukcje do ćwiczenia
- Utwórz instancję
WaterDatasetna podstawie plikuwater_train.csvi przypisz ją do zmiennejdataset_train. - Utwórz
dataloader_trainna podstawiedataset_train, używając rozmiaru partii równego dwa i włączając losowe mieszanie próbek. - Pobierz partię cech i etykiet z DataLoadera i wyświetl je.
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
# Create an instance of the WaterDataset
dataset_train = ____(____)
# Create a DataLoader based on dataset_train
dataloader_train = ____(
____,
batch_size=____,
shuffle=____,
)
# Get a batch of features and labels
features, labels = ____
print(features, labels)