PyTorch DataLoader
Ottimo lavoro nel definire la classe Dataset! Il WaterDataset che hai appena creato è ora pronto per essere usato.
Il passo successivo nella preparazione dei dati di training è impostare un DataLoader. Un DataLoader di PyTorch può essere creato a partire da un Dataset per caricare i dati, suddividerli in batch ed eventualmente applicare trasformazioni. Poi restituisce campioni pronti per l’addestramento.
In questo esercizio, costruirai un DataLoader basato su WaterDataset. La classe DataLoader di cui hai bisogno è già stata importata per te da torch.utils.data. Mettiamoci al lavoro!
Questo esercizio fa parte del corso
Deep Learning intermedio con PyTorch
Istruzioni dell'esercizio
- Crea un’istanza di
WaterDatasetdawater_train.csv, assegnandola adataset_train. - Crea
dataloader_traina partire dadataset_train, usando una dimensione di batch pari a due e mescolando i campioni. - Ottieni un batch di feature e label dal DataLoader e stampali.
Esercizio pratico interattivo
Prova a risolvere questo esercizio completando il codice di esempio.
# 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)