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 merupakan bagian dari kursus
Deep Learning Lanjutan dengan PyTorch
Instruksi 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 langsung praktik
Cobalah latihan ini dengan melengkapi kode contoh ini.
# 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)