IniziaInizia gratis

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

Visualizza il corso

Istruzioni dell'esercizio

  • Crea un’istanza di WaterDataset da water_train.csv, assegnandola a dataset_train.
  • Crea dataloader_train a partire da dataset_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)
Modifica ed esegui il codice