Zacznij terazZacznij za darmo

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

Zobacz kurs

Instrukcje do ćwiczenia

  • Utwórz instancję WaterDataset na podstawie pliku water_train.csv i przypisz ją do zmiennej dataset_train.
  • Utwórz dataloader_train na podstawie dataset_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)
Edytuj i uruchom kod