MulaiMulai sekarang secara gratis

PyTorch DataLoader

Kerja bagus mendefinisikan kelas Dataset! WaterDataset yang baru saja Anda buat sekarang sudah dapat digunakan.

Langkah berikutnya dalam menyiapkan data pelatihan adalah membuat DataLoader. Sebuah DataLoader di PyTorch dapat dibuat dari Dataset untuk memuat data, membaginya menjadi batch, dan melakukan transformasi pada data jika diperlukan. Setelah itu, DataLoader akan menghasilkan sampel data yang siap untuk pelatihan.

Dalam latihan ini, Anda akan membangun DataLoader berdasarkan WaterDataset. Kelas DataLoader yang Anda perlukan sudah diimpor untuk Anda dari torch.utils.data. Mari kita mulai!

Latihan ini adalah bagian dari kursus

Deep Learning Lanjutan dengan PyTorch

Lihat Kursus

Petunjuk latihan

  • Buat sebuah instance WaterDataset dari water_train.csv, lalu tetapkan ke dataset_train.
  • Buat dataloader_train berdasarkan dataset_train, dengan ukuran batch dua dan mengacak sampel.
  • Ambil satu batch fitur dan label dari DataLoader lalu cetak.

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

# 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)
Edit dan Jalankan Kode