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
Petunjuk latihan
- Buat sebuah instance
WaterDatasetdariwater_train.csv, lalu tetapkan kedataset_train. - Buat
dataloader_trainberdasarkandataset_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)